| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | function_saved_model_utils.py | 49 def _export_to_saved_model_graph(self, tensor_map, **unused_kwargs): argument
99 def __init__(self, function, tensor_map): argument
111 def _map_captures_to_created_tensors(original_captures, tensor_map, function): argument
|
| H A D | function.py | 2362 def _export_to_saved_model_graph(self, object_map, tensor_map, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantize_weights.cc | 213 absl::flat_hash_map<int32_t, TensorPerChannel>* tensor_map, in InsertQuantizableInputTensorsFromOperator()
407 absl::flat_hash_map<int32_t, TensorPerChannel> tensor_map; local
532 absl::flat_hash_map<int32_t, TensorT*> tensor_map; in QuantizeWeightsFloat16() local
|
| H A D | quantize_weights_portable.cc | 199 absl::flat_hash_map<int32_t, TensorPerChannel>* tensor_map, in InsertQuantizableInputTensorsFromOperator()
393 absl::flat_hash_map<int32_t, TensorPerChannel> tensor_map; local
518 absl::flat_hash_map<int32_t, TensorT*> tensor_map; in QuantizeWeightsFloat16() local
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/protocol/simple_aggregation/ |
| H A D | simple_aggregation_protocol.cc | 264 TensorMap tensor_map; in ParseCheckpoint() local
282 SimpleAggregationProtocol::TensorMap tensor_map) { in AggregateClientInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/trackable/ |
| H A D | asset.py | 106 def _export_to_saved_model_graph(self, tensor_map, **unused_kwargs): argument
|
| H A D | base.py | 1047 tensor_map=None, argument
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_scan.cpp | 89 TensorMap<Tensor<int, 1, DataLayout> > tensor_map(inputs, 20); in test_tensor_maps() local
|
| H A D | cxx11_tensor_io.cpp | 112 TensorMap<Tensor<const int, 1, DataLayout> > tensor_map(tensor.data(), 5); in test_output_const() local
|
| H A D | cxx11_tensor_reduction.cpp | 336 TensorMap<Tensor<int, 4, DataLayout> > tensor_map(inputs, 2, 3, 5, 7); in test_tensor_maps() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | models.py | 73 def _make_new_nodes(nodes_by_depth, layer_fn, layer_map, tensor_map): argument
|
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/python/ |
| H A D | PyQnnManagerAdaptor.h | 161 std::unordered_map<void*, int> tensor_map; in Compile() local
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/ |
| H A D | ClWorkloadRuntime.cpp | 203 std::map<ITensorInfo::Id, CLTensor *> tensor_map; in create_tensor_lut() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/registration/ |
| H A D | registration_saving_test.py | 54 def _export_to_saved_model_graph(self, object_map, tensor_map, **kwargs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | interpreter_builder.cc | 134 const flatbuffers::Vector<flatbuffers::Offset<tflite::TensorMap>>* in GetMapFromTensorMap()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | eigen_spatial_convolutions_test.cc | 1481 ArgType tensor_map(inputs[i].data(), input_dims); in PackRhsHelper() local
1686 ArgType tensor_map(filters[i].data(), filter_dims); in PackLhsHelper() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | profiler_python.cpp | 350 const py::dict& tensor_map) { in unpackTensorMap()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration_test.cpp | 1218 std::unordered_map<int64_t, Tensor> tensor_map; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 1098 const auto& tensor_map = data.tensor.flat<T>(); in GetSpanForData() local
|