Home
last modified time | relevance | path

Searched defs:shape_and_slices (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsave_restore_v2_ops.cc45 const Tensor& shape_and_slices) { in ValidateInputs()
100 const Tensor& shape_and_slices = context->input(2); in Compute() local
195 const Tensor& shape_and_slices = context->input(2); in Compute() local
H A Drestore_v2_op_test.cc120 Tensor shape_and_slices = MakeInput<tstring>( in RunTest() local
H A Dsave_restore_tensor.cc366 const Tensor& shape_and_slices, in RestoreTensorsV2()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/
H A Dio_ops.cc52 AbstractTensorHandle* const shape_and_slices, in RestoreV2()
77 AbstractTensorHandle* const shape_and_slices, in SaveV2()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dquantize_training.cc191 Tensor shape_and_slices; in ConnectVariablesToSaveOp() local
283 Node* shape_and_slices; in AddRestoreVariableSubgraphs() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_sharded_prefix_kernel.cc39 const Tensor* shape_and_slices; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/
H A Drestore.py100 def value_tensors(self, shape_and_slices=None): argument