Home
last modified time | relevance | path

Searched refs:GetTensorShapeMap (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dcheckpoint_reader.h49 const TensorShapeMap& GetTensorShapeMap() const { return shape_map_; } in GetTensorShapeMap() function
H A Dcheckpoint_writer_test.cc68 EXPECT_THAT(checkpoint_reader->GetTensorShapeMap(), in TEST()
H A Dcheckpoint_reader_test.cc58 EXPECT_THAT(checkpoint_reader->GetTensorShapeMap(), in TEST()