Searched defs:tensor_names (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | introduce_floating_point_jitter_pass_test.cc | 52 std::vector<string> tensor_names; in TEST() local
93 std::vector<string> tensor_names; in TEST() local
130 std::vector<string> tensor_names; in TEST() local
162 std::vector<string> tensor_names; in TEST() local
|
| D | introduce_floating_point_jitter_pass.cc | 28 const Graph& g, absl::Span<const string> tensor_names) { in GetNodesToModify()
123 absl::Span<string const> tensor_names, in IntroduceFloatingPointJitter()
|
| D | flags.h | 134 std::vector<string> tensor_names; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 42 const Tensor& prefix, const Tensor& tensor_names, in ValidateInputs()
97 const Tensor& tensor_names = context->input(1); in Compute() local
172 const Tensor& tensor_names = context->input(1); in Compute() local
|
| D | save_restore_tensor.cc | 337 const Tensor& tensor_names, in RestoreTensorsV2()
|
| D | restore_v2_op_test.cc | 66 const std::vector<string> tensor_names = { in RunTest() local
|
| D | restore_op_test.cc | 66 const std::vector<string> tensor_names = { in TEST_F() local
|
| D | remote_fused_graph_execute_utils.cc | 87 const std::vector<string>& tensor_names) { in ContainsSameTensorId()
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 37 std::unordered_set<string>* tensor_names) { in GetTensorNamesFromTensorInfo()
132 std::vector<string> tensor_names; in GetVariableNameToTensorMap() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | concat_xy.cc | 31 std::vector<std::string> tensor_names(op_def.src_tensors.size()); in GetConcatKernelCode() local
|
| D | concat_z.cc | 39 std::vector<std::string> tensor_names(op_def.src_tensors.size()); in GetConcatKernelCode() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 160 def get_tensors_from_tensor_names(graph, tensor_names): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | quantize_training.cc | 190 Tensor tensor_names; in ConnectVariablesToSaveOp() local
274 Node* tensor_names; in AddRestoreVariableSubgraphs() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 777 const std::vector<std::string>& tensor_names) { in GetTensorIndices()
805 auto tensor_names = llvm::map_range( in BuildTFEntryFunctionAttribute() local
|