Home
last modified time | relevance | path

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

/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dconverters_test.cc49 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()