| /external/tensorflow/tensorflow/lite/kernels/ |
| D | cast.cc | 52 void copyCast(const FromT* in, ToT* out, int num_elements) { in copyCast()
58 void copyCast(const std::complex<float>* in, ToT* out, int num_elements) { in copyCast()
66 int num_elements) { in copyCast()
73 int num_elements) { in copyToTensor()
104 const int num_elements = NumElements(input); in Eval() local
|
| D | neg.cc | 39 void Negate(const T* in_data, int num_elements, T* out_data) { in Negate()
49 const int num_elements = NumElements(input); in Eval() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | iota_test.cc | 27 std::vector<T> GetR1Expected(const int64 num_elements) { in GetR1Expected()
40 const int64 num_elements = std::get<1>(spec); in TEST_P() local
69 const int64 num_elements = std::get<1>(spec); in TEST_P() local
96 const int64 num_elements = std::get<1>(spec); in TEST_P() local
121 const int64 num_elements = 2; in TEST_P() local
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | stats_dataset_serialization_test.py | 36 def _build_dataset_bytes_stats(self, num_elements): argument
57 def _build_dataset_latency_stats(self, num_elements, tag="record_latency"): argument
62 num_elements, argument
|
| D | rebatch_dataset_serialization_test.py | 31 def build_dataset(num_elements, batch_size): argument
|
| D | unique_dataset_serialization_test.py | 31 def build_dataset(num_elements, unique_elem_range): argument
|
| D | scan_dataset_serialization_test.py | 29 def _build_dataset(self, num_elements): argument
|
| D | optimize_dataset_serialization_test.py | 31 def build_dataset(num_elements, batch_size): argument
|
| D | take_while_dataset_serialization_test.py | 32 def _build_dataset(self, num_elements, upper_bound): argument
|
| D | shard_dataset_serialization_test.py | 31 def _build_dataset(self, num_elements, num_shards, index): argument
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | benchmark_base.py | 33 def run_benchmark(self, dataset, num_elements, iters=1): argument
78 num_elements, argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.cc | 65 const std::size_t num_elements = shape.num_elements(); in FeatureDenseCopy() local
121 const int64 num_elements = values.value_size(); in FeatureSparseCopy() local
129 const int64 num_elements = values.value_size(); in FeatureSparseCopy() local
137 const int64 num_elements = values.value_size(); in FeatureSparseCopy() local
153 const int64 num_elements = in.shape().num_elements(); in CopyIntoSparseTensor() local
191 const std::size_t num_elements = in.shape().num_elements(); in RowDenseCopy() local
306 const int64 num_elements = t.shape().num_elements(); in GetSparseTensorShapes() local
397 const int64 num_elements = CopyIntoSparseTensor( in BatchExampleProtoToTensors() local
|
| D | example_proto_fast_parsing.cc | 99 bool GetNumElementsInBytesList(int* num_elements) { in GetNumElementsInBytesList()
475 LimitedArraySlice(T* begin, size_t num_elements) in LimitedArraySlice()
615 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample() local
668 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSerializedExample() local
825 const std::size_t num_elements = in.shape().num_elements(); in FastParseSerializedExample() local
911 const int d, const size_t num_elements, in FillAndCopyVarLen()
1186 const size_t num_elements = values.NumElements(); in FastParseExample() local
1364 const std::size_t num_elements = config.dense[d].elements_per_stride; in FastParseSingleExample() local
1407 size_t num_elements; in FastParseSingleExample() local
1579 int num_elements = 0; in ParseBytesFeature() local
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | parallel_interleave_benchmark.py | 45 def make_dataset(time_us, num_elements): argument
57 def _benchmark(self, dataset_fn, iters, num_elements): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | take_while_test.py | 38 def testTakeWhileDataset(self, num_elements, window_size): argument
54 def testTakeWhileDatasetRange(self, num_elements, upper_bound, out_of_bounds): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator.cc | 28 static string GetCompHloText(size_t num_elements) { in GetCompHloText()
85 int64 num_elements = ref_buffer.ElementCount(); in Create() local
131 int64 num_elements = test_buffer.ElementCount(); in CompareEqualImpl() local
|
| D | scratch_allocator.h | 45 int64 num_elements) { in Allocate()
|
| /external/libcxx/test/support/ |
| D | unique_ptr_test_helper.h | 39 newValue(int num_elements) { in newValue()
47 newValue(int num_elements) { in newValue()
|
| /external/eigen/test/ |
| D | umeyama.cpp | 91 void run_test(int dim, int num_elements) in run_test()
121 void run_fixed_size_test(int num_elements) in run_fixed_size_test()
162 const int num_elements = internal::random<int>(40,500); in test_umeyama() local
|
| /external/webrtc/webrtc/common_audio/ |
| D | ring_buffer_unittest.cc | 33 static int SetIncrementingData(int* data, int num_elements, in SetIncrementingData()
41 static int CheckIncrementingData(int* data, int num_elements, in CheckIncrementingData()
71 const int num_elements = rand() % buffer_size; in RandomStressTest() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dequantize_op.cc | 82 const int64 num_elements = input.NumElements(); in Compute() local
106 const int64 num_elements = input.NumElements(); in Compute() local
|
| /external/mesa3d/src/gallium/state_trackers/va/ |
| D | buffer.c | 40 unsigned int size, unsigned int num_elements, void *data, in vlVaCreateBuffer()
76 unsigned int num_elements) in vlVaBufferSetNumElements()
220 unsigned int *size, unsigned int *num_elements) in vlVaBufferInfo()
|
| /external/mesa3d/src/gallium/auxiliary/util/ |
| D | u_idalloc.h | 40 unsigned num_elements; member
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_testspec.h | 53 int64_t num_elements = 1; in GenerateRandomTensor() local
|
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | weighted_picker.h | 128 inline int WeightedPicker::num_elements() const { return N_; } in num_elements() function
|