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()
|