Searched refs:num_elements (Results 126 - 150 of 335) sorted by relevance

1234567891011>>

/external/tensorflow/tensorflow/core/kernels/
H A Dsdca_ops.cc299 const int64 num_elements = input.NumElements(); variable
301 0, TensorShape({num_elements, 2}), &out));
306 for (int64 i = 0; i < num_elements; ++i) {
H A Dlookup_table_op.h230 const int64 num_elements = table_->size(); variable
231 return num_elements * (sizeof(K) + sizeof(V));
H A Dsegment_reduction_ops_gpu.cu.cc140 if (data_size == 0 || segment_ids_shape.num_elements() == 0) {
184 if (data_size == 0 || segment_ids_shape.num_elements() == 0) {
H A Dscatter_nd_op.cc342 if (!(params_shape.num_elements() > 0 ||
396 *num_updates = indices_shape.num_elements() / safe_slice_dim;
456 if (shape.num_elements() == 0) {
466 out->shaped<T, 2>({shape.num_elements() / slice_size, slice_size});
470 if (shape.num_elements() > 0) {
/external/libmojo/mojo/public/cpp/bindings/lib/
H A Dnative_struct_serialization.h60 header->num_elements = static_cast<uint32_t>(pickle.payload_size());
76 // [num_bytes (4 bytes)] [num_elements (4 bytes)] [elements...]
/external/mesa3d/src/gallium/auxiliary/util/
H A Du_tests.c130 unsigned num_elements)
134 calloc(1, num_elements * sizeof(struct pipe_vertex_element));
136 for (i = 0; i < num_elements; i++) {
141 cso_set_vertex_elements(cso, num_elements, velem);
129 util_set_interleaved_vertex_elements(struct cso_context *cso, unsigned num_elements) argument
/external/mesa3d/src/gallium/drivers/freedreno/
H A Dfreedreno_state.c309 fd_vertex_state_create(struct pipe_context *pctx, unsigned num_elements, argument
317 memcpy(so->pipe, elements, sizeof(*elements) * num_elements);
318 so->num_elements = num_elements;
/external/mesa3d/src/gallium/drivers/nouveau/nv30/
H A Dnv30_state.h140 unsigned num_elements; member in struct:nv30_vertex_stateobj
/external/mesa3d/src/gallium/state_trackers/va/
H A Dva_private.h226 unsigned int num_elements; member in struct:__anon16887
306 unsigned int num_elements, void *data, VABufferID *buf_id);
307 VAStatus vlVaBufferSetNumElements(VADriverContextP ctx, VABufferID buf_id, unsigned int num_elements);
354 unsigned int *size, unsigned int *num_elements);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dscan_ops.cc71 if (input_shape.num_elements() == 0) {
H A Dtile_ops.cc47 OP_REQUIRES(ctx, input_shape.dims() == multiples_shape.num_elements(),
H A Dreverse_sequence_op.cc52 seq_lens_shape.num_elements() == input_shape.dim_size(batch_dim_),
54 "), ", "(", seq_lens_shape.num_elements(),
H A Dshape_op.cc92 const int64 size = input_shape.num_elements();
116 OP_REQUIRES(ctx, dim_shape.num_elements() == 1,
H A Dsplit_op.cc147 split_size_shape.num_elements() == num_split,
153 split_size_shape.num_elements(), " elements"));
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
H A Dautoregressive.py140 `distribution_fn(sample0).event_shape.num_elements()` are both `None`.
150 num_steps = self._distribution0.event_shape.num_elements()
/external/tensorflow/tensorflow/contrib/memory_stats/python/kernel_tests/
H A Dmemory_stats_ops_test.py61 matrix_size_in_bytes = matrix_shape.num_elements() * dtype.size
/external/drm_hwcomposer/
H A Ddrmhwctwo.cpp276 uint32_t *num_elements, hwc2_layer_t *layers, int32_t *types) {
281 if (layers && num_changes < *num_elements)
283 if (types && num_changes < *num_elements)
289 *num_elements = num_changes;
418 uint32_t *num_elements,
424 unsupported(__func__, display_requests, num_elements, layers, layer_requests);
425 *num_elements = 0;
449 HWC2::Error DrmHwcTwo::HwcDisplay::GetReleaseFences(uint32_t *num_elements, argument
459 } else if (num_layers > *num_elements) {
460 ALOGW("Overflow num_elements
275 GetChangedCompositionTypes( uint32_t *num_elements, hwc2_layer_t *layers, int32_t *types) argument
417 GetDisplayRequests(int32_t *display_requests, uint32_t *num_elements, hwc2_layer_t *layers, int32_t *layer_requests) argument
[all...]
/external/boringssl/src/crypto/evp/
H A Devp_asn1.c270 // num_elements parses one SEQUENCE from |in| and returns the number of elements
272 static size_t num_elements(const uint8_t *in, size_t in_len) { function
313 switch (num_elements(*inp, (size_t)len)) {
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
H A Dstats_accumulator_ops.cc190 int64 num_gradient_elements = gradient_shape.num_elements();
199 int64 num_hessian_elements = hessian_shape.num_elements();
305 std::vector<float> new_gradients(gradients_shape.num_elements());
306 for (int j = 0; j < gradients_shape.num_elements(); ++j) {
309 std::vector<float> new_hessians(hessians_shape.num_elements());
310 for (int j = 0; j < hessians_shape.num_elements(); ++j) {
316 for (int j = 0; j < gradients_shape.num_elements(); ++j) {
320 for (int j = 0; j < hessians_shape.num_elements(); ++j) {
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
H A Dscan_dataset_op_test.py131 def _build_dataset(self, num_elements):
132 return dataset_ops.Dataset.from_tensors(1).repeat(num_elements).apply(
/external/tensorflow/tensorflow/contrib/lite/java/src/main/native/
H A Dtensor_jni.cc35 const int num_elements = env->GetArrayLength(array); local
36 size_t to_copy = num_elements * elementByteSize(type);
/external/tensorflow/tensorflow/core/kernels/neon/
H A Dneon_depthwise_conv_op.cc100 FastBoundsCheck(out_shape.num_elements(),
118 if (out_shape.num_elements() == 0) {
/external/tensorflow/tensorflow/core/util/
H A Dexample_proto_helper.h191 elements_per_stride.push_back(dense_shape.num_elements());
257 elements_per_stride.push_back(dense_shape.num_elements());
/external/mesa3d/src/gallium/drivers/radeonsi/
H A Dsi_descriptors.c100 unsigned num_elements,
106 assert(num_elements <= sizeof(desc->dirty_mask)*8);
108 desc->list = CALLOC(num_elements, element_dw_size * 4);
110 desc->num_elements = num_elements;
111 desc->dirty_mask = num_elements == 32 ? ~0u : (1u << num_elements) - 1;
118 *ce_offset += align(element_dw_size * num_elements * 4, 32);
124 for (i = 0; i < num_elements * element_dw_size / 8; i++)
165 unsigned list_size = desc->num_elements * des
97 si_init_descriptors(struct si_descriptors *desc, unsigned shader_userdata_index, unsigned element_dw_size, unsigned num_elements, const uint32_t *null_descriptor, unsigned *ce_offset) argument
[all...]
/external/protobuf/src/google/protobuf/
H A Darena.h436 static T* CreateArray(::google::protobuf::Arena* arena, size_t num_elements) { argument
437 GOOGLE_CHECK_LE(num_elements,
441 return static_cast<T*>(::operator new[](num_elements * sizeof(T)));
443 return arena->CreateInternalRawArray<T>(num_elements);
630 T* CreateInternalRawArray(size_t num_elements) { argument
631 GOOGLE_CHECK_LE(num_elements,
635 AllocateAligned(RTTI_TYPE_ID(T), sizeof(T) * num_elements));

Completed in 1457 milliseconds

1234567891011>>