| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | add.cc | 32 const std::string tensor_name = absl::StrCat("src_tensor_", i); in CreateUnequalAdd() local
44 const std::string tensor_name = absl::StrCat("src_tensor_", i); in CreateUnequalAdd() local
56 const std::string tensor_name = absl::StrCat("src_tensor_", i); in CreateUnequalAdd() local
76 const std::string tensor_name = absl::StrCat("src_tensor_", i); in CreateAdd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | tfrt_graph_execution_state.cc | 509 for (const std::string& tensor_name : exit_node.input()) { in FindLoopCondFromExitNode() local
526 for (const std::string& tensor_name : switch_node->input()) { in FindLoopCondFromExitNode() local
582 for (const std::string& tensor_name : callable_options.fetch()) { in PruneGraphDef() local
593 for (const std::string& tensor_name : callable_options.target()) { in PruneGraphDef() local
607 for (const std::string& tensor_name : callable_options.feed()) { in PruneGraphDef() local
645 for (const std::string& tensor_name : node->input()) { in PruneGraphDef() local
715 for (const auto& tensor_name : node.input()) { in EliminateRefVariablesFromV1ControlFlow() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | debug_graph_utils.cc | 105 string tensor_name = in InsertNodes() local
148 const string tensor_name = in InsertNodes() local
288 const string DebugNodeInserter::GetDebugNodeName(const string& tensor_name, in GetDebugNodeName()
302 const string& tensor_name, const std::vector<string>& debug_ops, in CreateCopyNode()
473 const DataType src_dt, const string& tensor_name, in CreateDebugNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sendrecv_ops.cc | 34 const string& tensor_name) { in GetRendezvousKeyPrefix()
69 string tensor_name; in SendOp() local
150 string tensor_name; in RecvOp() local
|
| H A D | save_restore_tensor.cc | 172 const string& tensor_name = tensor_name_t.flat<tstring>()(restore_index); in RestoreTensor() local
354 string tensor_name; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | gpu_operation.cc | 362 absl::Status GPUOperation::GetTensorDescriptor(const std::string& tensor_name, in GetTensorDescriptor()
381 void GPUOperation::AddSrcTensor(const std::string& tensor_name, in AddSrcTensor()
395 void GPUOperation::AddDstTensor(const std::string& tensor_name, in AddDstTensor()
538 const std::string tensor_name = "src_tensor_" + std::to_string(i); in CreateGpuOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | memory_types.cc | 123 static Node* Send(Graph* g, const string& tensor_name, in Send()
139 static Node* Recv(Graph* g, const string& tensor_name, in Recv()
192 const string tensor_name = GetTensorName(e); in EnsureMemoryTypes() local
|
| H A D | step_stats_collector.cc | 90 string tensor_name; in Done() local
97 string tensor_name; in Done() local
|
| H A D | debugger_state_interface.cc | 33 string tensor_name = in SummarizeDebugTensorWatches() local
|
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_structured_sparsifier.py | 46 def update_mask(self, module, tensor_name, **kwargs): argument
51 def update_mask(self, module, tensor_name, **kwargs): argument
64 def update_mask(self, module, tensor_name, **kwargs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | checkpoint_utils.py | 458 tensor_name, argument
499 tensor_name): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | worker_cache_logger.cc | 87 const string& tensor_name, in RecordRecvTensor()
97 const string& tensor_name, in RecordDataTransfer()
|
| H A D | message_wrappers.cc | 825 const string& InMemoryRunStepResponse::tensor_name(size_t i) const { in tensor_name() function in tensorflow::InMemoryRunStepResponse
872 const string& OwnedProtoRunStepResponse::tensor_name(size_t i) const { in tensor_name() function in tensorflow::OwnedProtoRunStepResponse
928 const string& NonOwnedProtoRunStepResponse::tensor_name(size_t i) const { in tensor_name() function in tensorflow::NonOwnedProtoRunStepResponse
|
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/wrappers/ |
| H A D | TensorWrapper.cpp | 66 const std::string& tensor_name, in TensorWrapper()
161 const std::string& tensor_name, in CreateTensorWrapper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | dtensor_send_recv.cc | 213 std::string tensor_name = dtensor_send.key().str(); in LowerDTensorSendFromCPUToTFOp() local
237 std::string tensor_name = dtensor_recv.key().str(); in LowerDTensorRecvFromCPUToTFOp() local
|
| H A D | move_compilation_to_host.cc | 101 std::string tensor_name = absl::StrCat(kSendRecvKeyPrefix, *num_send_recv); in CreateSendRecvOpsToTransferProgramKey() local
110 const std::string& tensor_name = device_key_map[i]; in CreateSendRecvOpsToTransferProgramKey() local
|
| /aosp_15_r20/external/federated-compute/fcp/client/ |
| H A D | lc_runner.cc | 83 std::string tensor_name = input_resource_tensor_name_map[resource.first]; in ConstructInputsForTensorflowSpecPlan() local
119 std::string tensor_name = input_resource_tensor_name_map[resource.first]; in ConstructInputsForTFLitePlan() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | grappler_item_builder_test.cc | 46 Output tensor_name = in TEST_F() local
107 Output tensor_name = in TEST_F() local
|
| /aosp_15_r20/external/federated-compute/fcp/tensorflow/ |
| H A D | tensor_name.py | 28 def tensor_name(tensor): function
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/pruner/ |
| H A D | saliency_pruner.py | 15 def update_mask(self, module, tensor_name, **kwargs): argument
|
| H A D | lstm_saliency_pruner.py | 28 def update_mask(self, module, tensor_name, **kwargs): argument
|
| H A D | FPGM_pruner.py | 78 def update_mask(self, module, tensor_name, sparsity_level, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/sparsifier/ |
| H A D | nearly_diagonal_sparsifier.py | 36 def update_mask(self, module, tensor_name, nearliness, **kwargs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/protobuf/ |
| H A D | projector_config.proto | 31 string tensor_name = 1; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/ |
| H A D | inspect_checkpoint.py | 54 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
|