| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | rpcbench_test.cc | 124 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| H A D | calibrator_test.cc | 74 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
|
| H A D | calibration_logger.cc | 26 TfLiteStatus MinMax::Update(const float* values, size_t tensor_size, in Update()
|
| H A D | calibration_logger.h | 53 const float* tensor_values, size_t tensor_size, in LogTensorValue()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
| H A D | equality_assignment.h | 57 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
|
| H A D | greedy_in_order_assignment.h | 77 TensorSizeT tensor_size = usage_records[i].tensor_size; variable
171 const TensorSizeT& tensor_size = usage_records[i].tensor_size; in GreedyInOrderAssignmentMultidimensional() local
|
| H A D | types.h | 36 TensorSizeT tensor_size; member
|
| H A D | greedy_by_size_assignment.cc | 60 size_t tensor_size; member
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| H A D | constraint.py | 217 def __init__(self, tensor_size, input_var, dim_replace, index, output): argument
258 def __init__(self, tensor_size, input_var, index1, index2, output): argument
299 def __init__(self, tensor_size, index, res, input_var): argument
328 def __init__(self, tensor_size, index_tuple, res, input_var): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/inputs/ |
| H A D | trivial_test_graph_input_yielder.cc | 31 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()
|
| /aosp_15_r20/external/pytorch/binaries/ |
| H A D | record_function_benchmark.cc | 35 float runTensorGEMMBench(int tensor_size, int iter) { in runTensorGEMMBench()
66 for (auto tensor_size : std::set<int>({kSmallTensorSize, kTensorSize})) { in runBenchmark() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | unary_ops_composition_test.cc | 91 static Graph* UnaryOpsChain(int tensor_size, int repeat_graph, in UnaryOpsChain()
123 static Graph* UnaryOpsCompo(int tensor_size, int repeat_graph, in UnaryOpsCompo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/ |
| H A D | device_event_mgr_test.cc | 197 size_t tensor_size() const { in tensor_size() function in tensorflow::__anon437c77060511::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()
|
| /aosp_15_r20/external/pytorch/torch/multiprocessing/ |
| H A D | reductions.py | 125 tensor_size, argument
157 tensor_size, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | memory_planner.cpp | 296 auto tensor_size = ms.first; in allocateOutputTensors() local
411 auto tensor_size = storages_nbytes_[storages_idx]; in allocateManagedTensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate.cc | 543 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
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | prune.py | 1264 def _validate_pruning_amount(amount, tensor_size): argument
1309 def _compute_nparams_toprune(amount, tensor_size): argument
|
| /aosp_15_r20/external/XNNPACK/src/xnnpack/ |
| H A D | memory-planner.h | 26 size_t tensor_size; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_flatten.cpp | 21 size_t tensor_size = 0; in take_tensors() local
|
| /aosp_15_r20/external/XNNPACK/src/ |
| H A D | memory-planner.c | 138 size_t tensor_size) { in xnn_add_value_allocation_tracker()
|
| /aosp_15_r20/external/pytorch/torch/utils/tensorboard/ |
| H A D | _pytorch_graph.py | 42 tensor_size=None, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | dtype_analysis.cpp | 42 auto tensor_size = std::vector<int64_t>(rank.value(), 1); in MTensorArgumentCreator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | ring_alg.cc | 141 const size_t tensor_size = col_params->instance.shape.num_elements() * in GenerateSubdivsInCollectiveParams() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Slice.cpp | 38 uvec4 tensor_size; // output tensor size in slice_4d() member
|
| /aosp_15_r20/external/armnn/src/armnnUtils/ |
| H A D | TensorUtils.cpp | 82 auto tensor_size = tensorHandle->GetShape().GetNumElements(); in FindMinMax() local
|