Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_logger.h27 void Update(const float* values, size_t tensor_size) { in Update()
68 size_t tensor_size) { in LogTensorValue()
Dcalibrator_test.cc71 const size_t tensor_size = 1 * 8 * 8 * 3; in TEST() local
134 const size_t tensor_size = 1 * 8 * 8 * 3; in 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()
169 static void BM_Helper(int iters, int width, int num_stages, int tensor_size, in BM_Helper()
235 static void BM_RPC(int iters, int width, int tensor_size) { in BM_RPC()
/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/core/kernels/
Dunary_ops_composition_test.cc90 static Graph* UnaryOpsChain(int tensor_size, int repeat_graph, in UnaryOpsChain()
118 static Graph* UnaryOpsCompo(int tensor_size, int repeat_graph, in UnaryOpsCompo()
Dstack.cc166 const Tensor* tensor_size; in Compute() local
Dtensor_array_ops.cc184 const Tensor* tensor_size; in CreateTensorArray() local
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py189 def resize_tensor_input(self, input_index, tensor_size): argument
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc182 const int tensor_size = static_cast<int>(val.TotalBytes()); in AddRequest() local
/external/tensorflow/tensorflow/core/common_runtime/
Dring_alg.cc125 const size_t tensor_size = col_params->instance.shape.num_elements() * in GenerateSubdivsInCollectiveParams() local
Dring_reducer_test.cc633 size_t tensor_size = chunk_size * num_chunks; in TEST_F() local
/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.cc92 size_t tensor_size = tensor.bytes / sizeof(T); in TypedCheck() local
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc223 const size_t tensor_size = TF_TensorByteSize(src); in CopyTF_TensorStringsToPyArray() local
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle_test.cc887 int tensor_size) { in BM_BundleAlignmentByteOff()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc1046 const auto tensor_size = in GenerateRunMetadata() local
Dop_level_cost_estimator.cc1034 int64 tensor_size = 1; in CalculateTensorElementCount() local
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1534 size_t tensor_size = result_tensor_->TotalBytes(); in AllocateTensors() local
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2345 const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)}; in TEST_F() local