Home
last modified time | relevance | path

Searched refs:tensor_shapes_and_slices_t (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsave_restore_tensor.cc67 const Tensor& tensor_shapes_and_slices_t = context->input(2); in SaveTensors() local
70 tensor_shapes_and_slices_t.NumElements() == static_cast<int64_t>(N), in SaveTensors()
74 tensor_shapes_and_slices_t.NumElements())); in SaveTensors()
76 tensor_shapes_and_slices_t.flat<tstring>().data(); in SaveTensors()
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dappend_slices_op.cc106 const tensorflow::Tensor& tensor_shapes_and_slices_t = context->input(2); in SaveTensors() local
109 tensor_shapes_and_slices_t.NumElements() == static_cast<int64_t>(N), in SaveTensors()
114 tensor_shapes_and_slices_t.NumElements())); in SaveTensors()
116 tensor_shapes_and_slices_t.flat<tensorflow::tstring>().data(); in SaveTensors()