Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_message.proto38 string tensor_name = 4; field
60 string tensor_name = 2; field
/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.cc159 const string& tensor_name = tensor_name_t.flat<string>()(restore_index); in RestoreTensor() local
308 string tensor_name; member
344 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
361 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
396 const string& tensor_name = tensor_names_flat(i); in RestoreTensorsV2() local
/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/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()
440 const DataType src_dt, const string& tensor_name, in CreateDebugNode()
/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/python/training/
Dcheckpoint_utils.py291 tensor_name, argument
332 tensor_name): argument
/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/core/distributed_runtime/
Dworker_cache_logger.cc87 const string& tensor_name, in RecordRecvTensor()
97 const string& tensor_name, in RecordDataTransfer()
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc121 static Node* Send(Graph* g, const string& tensor_name, in Send()
135 static Node* Recv(Graph* g, const string& tensor_name, in Recv()
186 const string tensor_name = GetTensorName(e); in EnsureMemoryTypes() local
Ddebugger_state_interface.cc33 string tensor_name = in SummarizeDebugTensorWatches() local
Dkernel_benchmark_testlib.cc122 string tensor_name; in GetRendezvousKey() local
/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/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/contrib/model_pruning/python/
Dstrip_pruning_vars_lib.py37 def _node_name(tensor_name): argument
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform.py45 def _op_name(tensor_name): argument
635 def _is_removed(tensor_name, removed_op_names): argument
672 def _check_tensor_not_removed(tensor_name, removed_op_names): argument
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc34 StringPiece tensor_name, in CopyDeviceTensorToCPU()
/external/tensorflow/tensorflow/lite/experimental/micro/
Dsimple_tensor_allocator.cc114 const char* tensor_name = flatbuffer_tensor.name()->c_str(); in AllocateTensor() 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/python/framework/
Dimporter.py42 def _ParseTensorName(tensor_name): argument
/external/tensorflow/tensorflow/python/debug/cli/
Dcli_shared.py151 tensor_name, argument
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc55 const string& tensor_name, int dim1_size, in CopySubArrayToArray()

123