| /external/tensorflow/tensorflow/core/kernels/ |
| save_restore_tensor.h | 64 // * "tensor_names" and "shape_and_slices" shaped {N}, both DT_STRING.
68 const Tensor& shape_and_slices,
|
| save_restore_v2_ops.cc | 43 const Tensor& shape_and_slices) {
44 const int kFixedInputs = 3; // Prefix, tensor names, shape_and_slices.
52 TensorShapeUtils::IsVector(shape_and_slices.shape()),
54 "Input tensor_names and shape_and_slices "
57 shape_and_slices.shape().DebugString(), " instead."));
59 tensor_names.NumElements() == shape_and_slices.NumElements(),
60 errors::InvalidArgument("tensor_names and shape_and_slices "
63 shape_and_slices.NumElements()));
69 context, shape_and_slices.NumElements() == num_tensors,
98 const Tensor& shape_and_slices = context->input(2) variable
152 const Tensor& shape_and_slices = context->input(2); variable
[all...] |
| save_restore_tensor.cc | 231 const Tensor& shape_and_slices,
236 const auto& shape_and_slices_flat = shape_and_slices.flat<string>();
|
| restore_v2_op_test.cc | 54 .Input(FakeInput()) // shape_and_slices
77 .Input(FakeInput()) // shape_and_slices
117 Tensor shape_and_slices = MakeInput<string>( local
121 inputs.push_back({nullptr, &shape_and_slices});
|
| /external/tensorflow/tensorflow/core/graph/ |
| quantize_training.cc | 191 // Get the tensor_names and shape_and_slices tensors from the const op.
193 Tensor shape_and_slices; local
197 GetNodeAttr(shape_and_slices_op->attrs(), "value", &shape_and_slices));
220 FillStringTensor(&new_shape_and_slices, shape_and_slices);
267 strings::StrCat(new_restore_op_name, "/shape_and_slices");
278 // Construct the shape_and_slices input with empty string.
279 Node* shape_and_slices; local
285 .Finalize(graph, &shape_and_slices));
292 .Input(shape_and_slices)
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| checkpointable_utils.py | 419 shape_and_slices=[""],
|
| /external/tensorflow/tensorflow/python/training/ |
| checkpointable.py | 182 shape_and_slices=[""],
|
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
