| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 42 const Tensor& prefix, const Tensor& tensor_names, in ValidateInputs() argument
45 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in ValidateInputs()
51 TensorShapeUtils::IsVector(tensor_names.shape()) && in ValidateInputs()
56 tensor_names.shape().DebugString(), " and ", in ValidateInputs()
59 tensor_names.NumElements() == shape_and_slices.NumElements(), in ValidateInputs()
62 tensor_names.NumElements(), " vs. ", in ValidateInputs()
65 FastBoundsCheck(tensor_names.NumElements() + kFixedInputs, in ValidateInputs()
97 const Tensor& tensor_names = context->input(1); in Compute() local
99 ValidateInputs(true /* is save op */, context, prefix, tensor_names, in Compute()
103 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in Compute()
[all …]
|
| D | restore_v2_op_test.cc | 66 const std::vector<string> tensor_names = { in RunTest() local
116 TensorShape({static_cast<int>(tensor_names.size())}), in RunTest()
117 [&tensor_names](int x) -> string { return tensor_names[x]; }); in RunTest()
121 TensorShape({static_cast<int>(tensor_names.size())}), in RunTest()
204 [&](int x) -> tstring { return tensor_names[0]; }); in RunTest()
219 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[1]; in RunTest()
231 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[2]; in RunTest()
243 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[3]; in RunTest()
255 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[4]; in RunTest()
267 (*mutable_input(1).tensor).flat<tstring>()(0) = tensor_names[5]; in RunTest()
[all …]
|
| D | restore_op_test.cc | 66 const std::vector<string> tensor_names = { in TEST_F() local
105 TensorShape({static_cast<int>(tensor_names.size())}), in TEST_F()
106 [&tensor_names](int x) -> string { return tensor_names[x]; }); in TEST_F()
197 [&](int x) -> tstring { return tensor_names[0]; }); in TEST_F()
209 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[1]; in TEST_F()
221 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[2]; in TEST_F()
233 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[3]; in TEST_F()
245 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[4]; in TEST_F()
257 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[5]; in TEST_F()
270 (*mutable_input(1).tensor).scalar<tstring>()() = tensor_names[6]; in TEST_F()
[all …]
|
| D | save_restore_tensor.h | 67 const Tensor& tensor_names,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | introduce_floating_point_jitter_pass_test.cc | 52 std::vector<string> tensor_names; in TEST() local
53 tensor_names.push_back("sigmoid_a"); in TEST()
54 tensor_names.push_back("sigmoid_b"); in TEST()
56 TF_ASSERT_OK(IntroduceFloatingPointJitter(graph.get(), tensor_names, 0.01f)); in TEST()
93 std::vector<string> tensor_names; in TEST() local
94 tensor_names.push_back("sigmoid_a"); in TEST()
95 tensor_names.push_back("sigmoid_b"); in TEST()
97 TF_ASSERT_OK(IntroduceFloatingPointJitter(graph.get(), tensor_names, 0.01f)); in TEST()
130 std::vector<string> tensor_names; in TEST() local
131 tensor_names.push_back("sigmoid"); in TEST()
[all …]
|
| D | introduce_floating_point_jitter_pass.cc | 28 const Graph& g, absl::Span<const string> tensor_names) { in GetNodesToModify() argument
36 for (const string& tensor_name : tensor_names) { in GetNodesToModify()
123 absl::Span<string const> tensor_names, in IntroduceFloatingPointJitter() argument
125 if (tensor_names.empty()) { in IntroduceFloatingPointJitter()
131 GetNodesToModify(*graph, tensor_names); in IntroduceFloatingPointJitter()
150 return IntroduceFloatingPointJitter(options.graph->get(), flags.tensor_names, in Run()
|
| D | introduce_floating_point_jitter_pass_internal.h | 23 absl::Span<string const> tensor_names,
|
| D | flags.h | 133 std::vector<string> tensor_names; member
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 37 std::unordered_set<string>* tensor_names) { in GetTensorNamesFromTensorInfo() argument
42 tensor_names->insert(coo_sparse.values_tensor_name()); in GetTensorNamesFromTensorInfo()
43 tensor_names->insert(coo_sparse.indices_tensor_name()); in GetTensorNamesFromTensorInfo()
44 tensor_names->insert(coo_sparse.dense_shape_tensor_name()); in GetTensorNamesFromTensorInfo()
47 tensor_names->insert(component.name()); in GetTensorNamesFromTensorInfo()
50 tensor_names->insert(tensor_info.name()); in GetTensorNamesFromTensorInfo()
132 std::vector<string> tensor_names; in GetVariableNameToTensorMap() local
133 tensor_names.reserve(variable_names_set.size()); in GetVariableNameToTensorMap()
140 tensor_names.push_back(node_name + "/Read/ReadVariableOp:0"); in GetVariableNameToTensorMap()
142 tensor_names.push_back(node_name + ":0"); in GetVariableNameToTensorMap()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Save.pbtxt | 11 name: "tensor_names"
24 The size of `tensor_names` must match the number of tensors in `data`. `data[i]`
25 is written to `filename` with name `tensor_names[i]`.
|
| D | api_def_SaveSlices.pbtxt | 11 name: "tensor_names"
34 have as many elements as `tensor_names`.
|
| D | api_def_SaveV2.pbtxt | 11 name: "tensor_names"
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/CyclicModule/debug/ |
| D | saved_model_debug_info.pb | 82 5RestoreV2/tensor_names@__inference__traced_restore_45$
103 7RestoreV2_1/tensor_names@__inference__traced_restore_45$
131 /SaveV2/tensor_names@__inference__traced_save_30$
159 1SaveV2_1/tensor_names@__inference__traced_save_30$
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 64 tensor_names = []
69 tensor_names.append(spec.name)
73 return io_ops.save_v2(file_prefix, tensor_names, tensor_slices, tensors)
93 tensor_names, tensor_slices, tensor_dtypes = zip(*restore_specs)
96 file_prefix, tensor_names, tensor_slices, tensor_dtypes)
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/VarsAndArithmeticObjectGraph/debug/ |
| D | saved_model_debug_info.pb | 144 6RestoreV2/tensor_names@__inference__traced_restore_101(
165 8RestoreV2_1/tensor_names@__inference__traced_restore_101(
193 /SaveV2/tensor_names@__inference__traced_save_80(
235 1SaveV2_1/tensor_names@__inference__traced_save_80(
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Save.pbtxt | 8 name: "tensor_names"
29 name: "tensor_names"
|
| D | RestoreV2.pbtxt | 8 name: "tensor_names"
33 name: "tensor_names"
|
| D | SaveSlices.pbtxt | 8 name: "tensor_names"
33 name: "tensor_names"
|
| D | SaveV2.pbtxt | 8 name: "tensor_names"
33 name: "tensor_names"
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | quantize_training.cc | 192 Tensor tensor_names; in ConnectVariablesToSaveOp() local
195 GetNodeAttr(tensor_names_op->attrs(), "value", &tensor_names)); in ConnectVariablesToSaveOp()
199 int tn_size = tensor_names.NumElements(); in ConnectVariablesToSaveOp()
219 FillStringTensor(&new_tensor_names, tensor_names); in ConnectVariablesToSaveOp()
276 Node* tensor_names; in AddRestoreVariableSubgraphs() local
282 .Finalize(graph, &tensor_names)); in AddRestoreVariableSubgraphs()
297 .Input(tensor_names) in AddRestoreVariableSubgraphs()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | io_ops.py | 42 def _save(filename, tensor_names, tensors, tensor_slices=None, name="save"): argument
69 return gen_io_ops.save(filename, tensor_names, tensors, name=name)
71 return gen_io_ops.save_slices(filename, tensor_names, tensor_slices,
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/ |
| D | saved_model.pbtxt | 331 name: "tensor_names"
356 name: "tensor_names"
1828 name: "save/SaveV2/tensor_names"
1907 input: "save/SaveV2/tensor_names"
2023 name: "save/RestoreV2/tensor_names"
2098 input: "save/RestoreV2/tensor_names"
2161 name: "save/RestoreV2_1/tensor_names"
2236 input: "save/RestoreV2_1/tensor_names"
2299 name: "save/RestoreV2_2/tensor_names"
2374 input: "save/RestoreV2_2/tensor_names"
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_util_test.py | 532 tensor_names = [tensor.name for tensor in model.inputs + model.outputs]
538 tensor_names)
554 tensor_names = [tensor.name for tensor in model.inputs + model.outputs]
560 tensor_names)
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | util.py | 95 def get_tensors_from_tensor_names(graph, tensor_names): argument
118 for name in tensor_names:
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 104 tensor_names = []
109 tensor_names.append(spec.name)
115 tensor_names=tensor_names,
121 return io_ops.save_v2(filename_tensor, tensor_names, tensor_slices,
|