/external/tensorflow/tensorflow/core/kernels/ |
H A D | sdca_ops.cc | 299 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 D | lookup_table_op.h | 230 const int64 num_elements = table_->size(); variable 231 return num_elements * (sizeof(K) + sizeof(V));
|
H A D | segment_reduction_ops_gpu.cu.cc | 140 if (data_size == 0 || segment_ids_shape.num_elements() == 0) { 184 if (data_size == 0 || segment_ids_shape.num_elements() == 0) {
|
H A D | scatter_nd_op.cc | 342 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 D | native_struct_serialization.h | 60 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 D | u_tests.c | 130 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 D | freedreno_state.c | 309 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 D | nv30_state.h | 140 unsigned num_elements; member in struct:nv30_vertex_stateobj
|
/external/mesa3d/src/gallium/state_trackers/va/ |
H A D | va_private.h | 226 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 D | scan_ops.cc | 71 if (input_shape.num_elements() == 0) {
|
H A D | tile_ops.cc | 47 OP_REQUIRES(ctx, input_shape.dims() == multiples_shape.num_elements(),
|
H A D | reverse_sequence_op.cc | 52 seq_lens_shape.num_elements() == input_shape.dim_size(batch_dim_), 54 "), ", "(", seq_lens_shape.num_elements(),
|
H A D | shape_op.cc | 92 const int64 size = input_shape.num_elements(); 116 OP_REQUIRES(ctx, dim_shape.num_elements() == 1,
|
H A D | split_op.cc | 147 split_size_shape.num_elements() == num_split, 153 split_size_shape.num_elements(), " elements"));
|
/external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
H A D | autoregressive.py | 140 `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 D | memory_stats_ops_test.py | 61 matrix_size_in_bytes = matrix_shape.num_elements() * dtype.size
|
/external/drm_hwcomposer/ |
H A D | drmhwctwo.cpp | 276 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 D | evp_asn1.c | 270 // 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 D | stats_accumulator_ops.cc | 190 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 D | scan_dataset_op_test.py | 131 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 D | tensor_jni.cc | 35 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 D | neon_depthwise_conv_op.cc | 100 FastBoundsCheck(out_shape.num_elements(), 118 if (out_shape.num_elements() == 0) {
|
/external/tensorflow/tensorflow/core/util/ |
H A D | example_proto_helper.h | 191 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 D | si_descriptors.c | 100 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 D | arena.h | 436 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));
|