| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| 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
|
| D | concat_z.cc | 104 std::string tensor_name = "args." + tensor_names[i]; in GetConcatKernelCode() local
|
| /external/tensorflow/tensorflow/core/debug/ |
| 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()
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| 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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
|
| 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
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| 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
|
| D | debugger_state_interface.cc | 33 string tensor_name = in SummarizeDebugTensorWatches() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_cache_logger.cc | 87 const string& tensor_name, in RecordRecvTensor()
97 const string& tensor_name, in RecordDataTransfer()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 458 tensor_name, argument
499 tensor_name): argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| 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
|
| /external/tensorflow/tensorflow/python/keras/protobuf/ |
| D | projector_config.proto | 31 string tensor_name = 1; field
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | inspect_checkpoint.py | 54 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| 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
|
| 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
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 36 StringPiece tensor_name, in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_context.cc | 35 StringPiece tensor_name, in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_remaining_ops.cc | 152 const std::string& tensor_name) { in GetRendezvousKey()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 48 StringPiece tensor_name, in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 38 StringPiece tensor_name, in CopyDeviceTensorToCPUSync()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_matmul_op_gpu.cu.cc | 97 const std::string& tensor_name, in ValidateInputTensor()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_util_v2.py | 146 def resource_input_index(tensor_name, input_names, node_defs, functions): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| D | utils.cc | 60 Status GetTrtBindingIndex(const char* tensor_name, int profile_index, in GetTrtBindingIndex()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | introduce_floating_point_jitter_pass.cc | 36 for (const string& tensor_name : tensor_names) { in GetNodesToModify() local
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder_test.cc | 46 Output tensor_name = in TEST_F() local
107 Output tensor_name = in TEST_F() local
|