Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc29 const string& tensor_name) { in GetRendezvousKeyPrefix()
64 string tensor_name; in SendOp() local
132 string tensor_name; in RecvOp() local
Dsave_restore_tensor.cc157 const string& tensor_name = tensor_name_t.flat<string>()(restore_index); in RestoreTensor() local
255 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_message.proto38 string tensor_name = 4; field
60 string tensor_name = 2; field
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc78 string tensor_name = in InsertNodes() local
121 const string tensor_name = in InsertNodes() local
255 const string DebugNodeInserter::GetDebugNodeName(const string& tensor_name, in GetDebugNodeName()
269 const string& tensor_name, const std::vector<string>& debug_ops, in CreateCopyNode()
441 const DataType src_dt, const string& tensor_name, in CreateDebugNode()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.py36 def _add_input_to_signature_def(tensor_name, map_key, signature_def): argument
52 def _add_output_to_signature_def(tensor_name, map_key, signature_def): argument
Dmanifest.proto22 string tensor_name = 1; field
Dbundle_shim.cc244 const string& tensor_name, in AddInputToSignatureDef()
274 const string& tensor_name, in AddOutputToSignatureDef()
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py276 tensor_name, argument
299 tensor_name): argument
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dcheckpoint_utils.py107 def _set_checkpoint_initializer(variable, file_pattern, tensor_name, slice_spec, argument
125 tensor_name): argument
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc120 static Node* Send(Graph* g, const string& tensor_name, in Send()
134 static Node* Recv(Graph* g, const string& tensor_name, in Recv()
185 const string tensor_name = GetTensorName(e); in EnsureMemoryTypes() local
Ddebugger_state_interface.cc33 string tensor_name = in SummarizeDebugTensorWatches() local
Dkernel_benchmark_testlib.cc117 string tensor_name; in GetRendezvousKey() 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/python/debug/lib/
Dstepper.py392 def override_tensor(self, tensor_name, overriding_val): argument
419 def remove_override(self, tensor_name): argument
917 def get_tensor_value(self, tensor_name): argument
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc93 StringPiece tensor_name, in CopyDeviceTensorToCPU()
138 StringPiece tensor_name, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dinspect_checkpoint.py30 def print_tensors_in_checkpoint_file(file_name, tensor_name): argument
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform.py41 def _op_name(tensor_name): argument
625 def _is_removed(tensor_name, removed_op_names): argument
662 def _check_tensor_not_removed(tensor_name, removed_op_names): argument
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc33 StringPiece tensor_name, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/python/tools/
Dinspect_checkpoint.py32 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
/external/tensorflow/tensorflow/contrib/tensorboard/plugins/projector/
Dprojector_config.proto27 string tensor_name = 1; field
/external/tensorflow/tensorflow/python/framework/
Dimporter.py97 def _ParseTensorName(tensor_name): argument
760 def _IsImportedNodeOutput(tensor_name): argument
/external/tensorflow/tensorflow/python/estimator/
Dmodel_fn.py334 def _check_is_tensor(x, tensor_name): argument
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc136 Output tensor_name = in TEST_F() local
197 Output tensor_name = in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc326 const string tensor_name = AddPort(node_name); in FillInputNode() local
385 const string tensor_name = AddPort(node_name); in ReadOutputNode() local
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dlstm_utils.cc55 const string& tensor_name, int dim1_size, in CopySubArrayToArray()

123