Home
last modified time | relevance | path

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

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