Home
last modified time | relevance | path

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

123

/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/kernels/
Dsave_restore_tensor.cc159 const string& tensor_name = tensor_name_t.flat<tstring>()(restore_index); in RestoreTensor() local
327 string tensor_name; member
363 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
380 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
415 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
Dsave_restore_v2_ops.cc113 const string& tensor_name = tensor_names_flat(i); in Compute() local
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py405 tensor_name, argument
446 tensor_name): argument
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc121 static Node* Send(Graph* g, const string& tensor_name, in Send()
137 static Node* Recv(Graph* g, const string& tensor_name, in Recv()
190 const string tensor_name = GetTensorName(e); in EnsureMemoryTypes() local
Ddebugger_state_interface.cc33 string tensor_name = in SummarizeDebugTensorWatches() local
Dgraph_runner.cc134 const string& tensor_name = in.first; in Run() local
/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/python/tools/
Dinspect_checkpoint.py57 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
/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/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc36 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/kernels/hexagon/
Dhexagon_control_wrapper.cc325 const string tensor_name = AddPort(node_name); in FillInputNode() local
384 const string tensor_name = AddPort(node_name); in ReadOutputNode() local
/external/tensorflow/tensorflow/python/debug/cli/
Dcli_shared.py151 tensor_name, argument
/external/tensorflow/tensorflow/python/framework/
Dimporter.py42 def _ParseTensorName(tensor_name): argument
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc55 const std::string& tensor_name, int dim1_size, in CopySubArrayToArray()
/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/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc105 Status GetTrtBindingIndex(const char* tensor_name, int profile_index, in GetTrtBindingIndex()

123