| /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 | 158 std::vector<string> tensor_names; member
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | io_ops.cc | 51 AbstractTensorHandle* const tensor_names, in RestoreV2()
76 AbstractTensorHandle* const tensor_names, in SaveV2()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 44 const Tensor& prefix, const Tensor& tensor_names, in ValidateInputs()
99 const Tensor& tensor_names = context->input(1); in Compute() local
194 const Tensor& tensor_names = context->input(1); in Compute() local
|
| D | save_restore_tensor.cc | 365 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
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 38 std::unordered_set<string>* tensor_names) { in GetTensorNamesFromTensorInfo()
133 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 | 40 std::vector<std::string> tensor_names(op_def.src_tensors.size()); in GetConcatKernelCode() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | save_restore_spmd_expander.cc | 271 mlir::Value tensor_names = new_save.getArgument(1); in ConditionalSave() local
324 mlir::Value tensor_names = new_tensor_names[0]; in ConditionalSave() local
619 mlir::Value tensor_names = device_restore_fn.getArgument(1); in ExpandRestoreV2OpHelper() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_sharded_prefix_kernel.cc | 38 const Tensor* tensor_names; in Compute() local
|
| /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/python/ops/ |
| D | io_ops.py | 39 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 130 def get_tensors_from_tensor_names(graph, tensor_names): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 938 const std::vector<std::string>& tensor_names) { in GetTensorIndices()
966 auto tensor_names = llvm::map_range( in BuildTFEntryFunctionAttribute() local
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle_test.cc | 940 std::vector<string> tensor_names = {"tensor-1-0", "tensor-0-1", "tensor-1-2", in TEST() local
|