Home
last modified time | relevance | path

Searched defs:tensor_size (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dequality_assignment.h57 const TensorSizeT tensor_size = usage_records[i].tensor_size; in EqualityAssignmentWithHash() local
91 const TensorSizeT tensor_size = usage_records[i].tensor_size; in EqualityAssignment() local
Dgreedy_in_order_assignment.h77 TensorSizeT tensor_size = usage_records[i].tensor_size; variable
171 const TensorSizeT& tensor_size = usage_records[i].tensor_size; in GreedyInOrderAssignmentMultidimensional() local
Dtypes.h36 TensorSizeT tensor_size; member
Dgreedy_by_size_assignment.cc59 size_t tensor_size; member
/external/tensorflow/tensorflow/lite/micro/
Dtesting_helpers_test.cc24 constexpr int tensor_size = 12; in TF_LITE_MICRO_TEST() local
46 constexpr int tensor_size = 12; in TF_LITE_MICRO_TEST() local
76 const int tensor_size = 12; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc124 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef()
157 string DebugString(const Tensor& x, const Tensor& y, int tensor_size) { in DebugString()
170 int num_stages, int tensor_size, in BM_Helper()
238 const int tensor_size = state.range(1); in BM_RPC() local
/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.cc31 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef()
90 int num_stages, int width, int tensor_size, bool insert_queue, in TrivialTestGraphInputYielder()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_logger.cc26 TfLiteStatus MinMax::Update(const float* values, size_t tensor_size, in Update()
Dcalibration_logger.h53 size_t tensor_size, in LogTensorValue()
Dcalibrator_test.cc72 const size_t tensor_size = 1 * 8 * 8 * 3; in TEST() local
135 const size_t tensor_size = 1 * 8 * 8 * 3; in TEST() local
211 const size_t tensor_size = 1 * 8 * 8 * 3; in TEST() local
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition_test.cc91 static Graph* UnaryOpsChain(int tensor_size, int repeat_graph, in UnaryOpsChain()
123 static Graph* UnaryOpsCompo(int tensor_size, int repeat_graph, in UnaryOpsCompo()
Dstack.cc167 const Tensor* tensor_size; in Compute() local
Dtensor_array_ops.cc185 const Tensor* tensor_size; in CreateTensorArray() local
/external/XNNPACK/src/xnnpack/
Dmemory-planner.h26 size_t tensor_size; member
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc197 size_t tensor_size() const { in tensor_size() function in tensorflow::__anon9d5fcaf70511::EMBenchmarkHelper
206 void ReInit(int num_ops, int tensor_size) { in ReInit()
479 static void BM_chain_ops(int iters, int tensor_size, int adds_per_round, in BM_chain_ops()
/external/XNNPACK/src/
Dmemory-planner.c138 size_t tensor_size) { in xnn_add_value_allocation_tracker()
/external/tensorflow/tensorflow/core/common_runtime/
Dring_alg.cc126 const size_t tensor_size = col_params->instance.shape.num_elements() * in GenerateSubdivsInCollectiveParams() local
Dring_reducer_test.cc650 size_t tensor_size = chunk_size * num_chunks; in TEST_F() local
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py501 def resize_tensor_input(self, input_index, tensor_size, strict=False): argument
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtflite_driver.cc94 size_t tensor_size = tensor.bytes / sizeof(T); in TypedCheck() local
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver.cc174 size_t tensor_size = tensor.bytes / sizeof(T); in TypedCheck() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h577 int tensor_size; in ExtractVector() local
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle_test.cc1116 int alignment, int tensor_size) { in BM_BundleAlignmentByteOff()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc3874 int tensor_size = 0; in Invoke() local
3944 int tensor_size = 0; in Invoke() local
4710 int tensor_size = 0; in BuildGraph() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc1222 const auto tensor_size = in GenerateRunMetadata() local

12