Home
last modified time | relevance | path

Searched defs:tensor_size (Results 1 – 25 of 31) 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.cc60 size_t tensor_size; member
/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/
Dcalibrator_test.cc74 const size_t tensor_size = 1 * 8 * 8 * 3; in TEST() local
137 const size_t tensor_size = 1 * 8 * 8 * 3; in TEST() local
218 const size_t tensor_size = 1 * 8 * 8 * 3; in TEST() local
516 const size_t tensor_size = 1; in TEST() local
579 const size_t tensor_size = 1; in TEST() local
Dcalibration_logger.cc26 TfLiteStatus MinMax::Update(const float* values, size_t tensor_size, in Update()
Dcalibration_logger.h53 const float* tensor_values, size_t tensor_size, in LogTensorValue()
/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::__anon8b4f34150511::EMBenchmarkHelper
206 void ReInit(int num_ops, int tensor_size) { in ReInit()
492 static void BM_chain_ops(::testing::benchmark::State& state, int tensor_size, 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.cc141 const size_t tensor_size = col_params->instance.shape.num_elements() * in GenerateSubdivsInCollectiveParams() local
Dring_reducer_test.cc286 size_t tensor_size = chunk_size * num_chunks; in TEST_F() local
/external/armnn/src/armnnUtils/
DTensorUtils.cpp82 auto tensor_size = tensorHandle->GetShape().GetNumElements(); in FindMinMax() 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/python/
Dinterpreter.py699 def resize_tensor_input(self, input_index, tensor_size, strict=False): argument
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc543 static size_t GetNNTensorSize(size_t tensor_size, bool allow_padding) { in GetNNTensorSize()
4854 int tensor_size = 0; in Invoke() local
4881 size_t tensor_size = tensor.bytes; in Invoke() local
4950 int tensor_size = 0; in Invoke() local
5076 size_t tensor_size = tensor->bytes; in Invoke() local
5185 size_t tensor_size = tensor->bytes; in Invoke() local
6168 int tensor_size = 0; in BuildGraph() local
6194 size_t tensor_size = in BuildGraph() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc221 int64_t tensor_size = MultiplyWithoutOverflow(num_elems, size); in CalculateTensorSize() local
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBlock.h406 IndexType tensor_size = m_tensor_dimensions.TotalSize(); in InitializeBlockDimensions() local
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver.cc191 size_t tensor_size = tensor.bytes / sizeof(T); in TypedCheck() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h621 int tensor_size; in ExtractVector() local

12