Searched refs:CreateTfTensorSpec (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/ |
H A D | converters_test.cc | 49 tf::TensorSpecProto CreateTfTensorSpec( in CreateTfTensorSpec() function 81 ConvertTensorSpec(CreateTfTensorSpec("foo", tf::DT_FLOAT, {1, 2, 3})); in TEST() 90 ConvertTensorSpec(CreateTfTensorSpec("foo", tf::DT_VARIANT, {1, 2, 3})), in TEST() 96 ConvertTensorSpec(CreateTfTensorSpec("foo", tf::DT_FLOAT, {1, -1})), in TEST()
|