| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_v2_ops.cc | 43 const Tensor& shape_and_slices) { in ValidateInputs() argument
52 TensorShapeUtils::IsVector(shape_and_slices.shape()), in ValidateInputs()
57 shape_and_slices.shape().DebugString(), " instead.")); in ValidateInputs()
59 tensor_names.NumElements() == shape_and_slices.NumElements(), in ValidateInputs()
63 shape_and_slices.NumElements())); in ValidateInputs()
69 context, shape_and_slices.NumElements() == num_tensors, in ValidateInputs()
98 const Tensor& shape_and_slices = context->input(2); in Compute() local
100 shape_and_slices); in Compute()
106 const auto& shape_and_slices_flat = shape_and_slices.flat<string>(); in Compute()
152 const Tensor& shape_and_slices = context->input(2); in Compute() local
[all …]
|
| D | save_restore_tensor.h | 68 const Tensor& shape_and_slices,
|
| D | save_restore_tensor.cc | 319 const Tensor& shape_and_slices, in RestoreTensorsV2() argument
324 const auto& shape_and_slices_flat = shape_and_slices.flat<string>(); in RestoreTensorsV2()
|
| D | restore_v2_op_test.cc | 117 Tensor shape_and_slices = MakeInput<string>( in RunTest() local
121 inputs.push_back({nullptr, &shape_and_slices}); in RunTest()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SaveV2.pbtxt | 17 name: "shape_and_slices"
32 specific slices of full tensors, "shape_and_slices" should be non-empty strings
|
| D | api_def_RestoreV2.pbtxt | 16 name: "shape_and_slices"
47 specific slices of stored tensors, "shape_and_slices" should be non-empty
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | quantize_training.cc | 193 Tensor shape_and_slices; in ConnectVariablesToSaveOp() local
197 GetNodeAttr(shape_and_slices_op->attrs(), "value", &shape_and_slices)); in ConnectVariablesToSaveOp()
220 FillStringTensor(&new_shape_and_slices, shape_and_slices); in ConnectVariablesToSaveOp()
285 Node* shape_and_slices; in AddRestoreVariableSubgraphs() local
291 .Finalize(graph, &shape_and_slices)); in AddRestoreVariableSubgraphs()
298 .Input(shape_and_slices) in AddRestoreVariableSubgraphs()
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/ |
| D | saved_model.pbtxt | 335 name: "shape_and_slices"
360 name: "shape_and_slices"
1866 name: "save/SaveV2/shape_and_slices"
1908 input: "save/SaveV2/shape_and_slices"
2059 name: "save/RestoreV2/shape_and_slices"
2099 input: "save/RestoreV2/shape_and_slices"
2197 name: "save/RestoreV2_1/shape_and_slices"
2237 input: "save/RestoreV2_1/shape_and_slices"
2335 name: "save/RestoreV2_2/shape_and_slices"
2375 input: "save/RestoreV2_2/shape_and_slices"
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | base.py | 300 shape_and_slices=[""],
|
| D | util.py | 237 shape_and_slices=[""],
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3037 …argspec: "args=[\'prefix\', \'tensor_names\', \'shape_and_slices\', \'dtypes\', \'name\'], varargs…
3161 …argspec: "args=[\'prefix\', \'tensor_names\', \'shape_and_slices\', \'tensors\', \'name\'], vararg…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3037 …argspec: "args=[\'prefix\', \'tensor_names\', \'shape_and_slices\', \'dtypes\', \'name\'], varargs…
3161 …argspec: "args=[\'prefix\', \'tensor_names\', \'shape_and_slices\', \'tensors\', \'name\'], vararg…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12278 func SaveV2(scope *Scope, prefix tf.Output, tensor_names tf.Output, shape_and_slices tf.Output, ten…
12285 prefix, tensor_names, shape_and_slices, tf.OutputList(tensors),
31177 func RestoreV2(scope *Scope, prefix tf.Output, tensor_names tf.Output, shape_and_slices tf.Output, …
31185 prefix, tensor_names, shape_and_slices,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 27853 name: "shape_and_slices"
28564 name: "shape_and_slices"
|
| D | ops_history.v1.pbtxt | 63587 name: "shape_and_slices"
63612 name: "shape_and_slices"
65435 name: "shape_and_slices"
65460 name: "shape_and_slices"
|
| D | ops_history.v2.pbtxt | 62960 name: "shape_and_slices"
62985 name: "shape_and_slices"
64808 name: "shape_and_slices"
64833 name: "shape_and_slices"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 31367 name: "shape_and_slices"
32655 name: "shape_and_slices"
|