Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dtf_session_helper.cc682 TF_GraphSetTensorShape(graph, output, nullptr, -1, status); in TF_GraphSetTensorShape_wrapper()
685 TF_GraphSetTensorShape(graph, output, dims.data(), dims.size(), status); in TF_GraphSetTensorShape_wrapper()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_test.cc380 TF_GraphSetTensorShape(graph, feed_out_0, /*dims=*/nullptr, -1, s); in TEST()
388 TF_GraphSetTensorShape(graph, feed_out_0, dims, 2, s); in TEST()
405 TF_GraphSetTensorShape(graph, feed_out_0, dims, 2, s); in TEST()
416 TF_GraphSetTensorShape(graph, feed_out_0, /*dims=*/nullptr, -1, s); in TEST()
428 TF_GraphSetTensorShape(graph, feed_out_0, dims, 2, s); in TEST()
443 TF_GraphSetTensorShape(graph, feed_out_0, dims, 2, s); in TEST()
H A Dc_api.h207 TF_CAPI_EXPORT extern void TF_GraphSetTensorShape(TF_Graph* graph,
H A Dc_api.cc688 void TF_GraphSetTensorShape(TF_Graph* graph, TF_Output output, in TF_GraphSetTensorShape() function