Home
last modified time | relevance | path

Searched defs:tensor_name (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.cc201 for (const std::string& tensor_name : exit_node.input()) { in FindLoopCondFromExitNode() local
218 for (const std::string& tensor_name : switch_node->input()) { in FindLoopCondFromExitNode() local
274 for (const std::string& tensor_name : callable_options.fetch()) { in PruneGraphDef() local
285 for (const std::string& tensor_name : callable_options.target()) { in PruneGraphDef() local
299 for (const std::string& tensor_name : callable_options.feed()) { in PruneGraphDef() local
337 for (const std::string& tensor_name : node->input()) { in PruneGraphDef() local
407 for (const auto& tensor_name : node.input()) { in EliminateRefVariablesFromV1ControlFlow() local
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.cc166 const string& tensor_name = tensor_name_t.flat<tstring>()(restore_index); in RestoreTensor() local
334 string tensor_name; member
370 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
387 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
422 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
Dsendrecv_ops.cc32 const string& tensor_name) { in GetRendezvousKeyPrefix()
67 string tensor_name; in SendOp() local
146 string tensor_name; in RecvOp() local
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc105 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/common_runtime/
Dmemory_types.cc123 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
Ddebugger_state_interface.cc33 string tensor_name = in SummarizeDebugTensorWatches() local
Dkernel_benchmark_testlib.cc149 string tensor_name; in GetRendezvousKey() local
Dgraph_runner.cc134 const string& tensor_name = in.first; in Run() local
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py456 tensor_name, argument
497 tensor_name): argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache_logger.cc87 const string& tensor_name, in RecordRecvTensor()
97 const string& tensor_name, in RecordDataTransfer()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dadd.cc37 const std::string tensor_name = absl::StrCat("src_data_", i); in CreateAdd() local
Dconcat_z.cc90 std::string tensor_name = "args." + tensor_names[i]; in GetConcatKernelCode() local
/external/tensorflow/tensorflow/python/keras/protobuf/
Dprojector_config.proto31 string tensor_name = 1; field
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc158 void GPUOperation::AddSrcTensor(const std::string& tensor_name, in AddSrcTensor()
179 void GPUOperation::AddDstTensor(const std::string& tensor_name, in AddDstTensor()
/external/tensorflow/tensorflow/python/tools/
Dinspect_checkpoint.py58 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc36 StringPiece tensor_name, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_context.cc35 StringPiece tensor_name, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.cc48 StringPiece tensor_name, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.cc38 StringPiece tensor_name, in CopyDeviceTensorToCPUSync()
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py150 def resource_input_index(tensor_name, input_names, node_defs, functions): argument
/external/tensorflow/tensorflow/compiler/jit/
Dintroduce_floating_point_jitter_pass.cc36 for (const string& tensor_name : tensor_names) { in GetNodesToModify() local
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc46 Output tensor_name = in TEST_F() local
107 Output tensor_name = in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc55 const std::string& tensor_name, int dim1_size, in CopySubArrayToArray()
/external/tensorflow/tensorflow/python/debug/cli/
Dcli_shared.py151 tensor_name, argument
/external/tensorflow/tensorflow/python/framework/
Dimporter.py42 def _ParseTensorName(tensor_name): argument

123