Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dtpu_replication_ops.cc71 (void)c->MergeInputHandleShapesAndTypes(i, *shapes_and_types); in __anon6a17d95b0102()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.h654 bool MergeInputHandleShapesAndTypes(
H A Dshape_inference.cc1250 bool InferenceContext::MergeInputHandleShapesAndTypes( in MergeInputHandleShapesAndTypes() function in tensorflow::shape_inference::InferenceContext
H A Dshape_inference_test.cc1724 return c.MergeInputHandleShapesAndTypes(idx, shapes_and_types); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.cc365 node_context->MergeInputHandleShapesAndTypes(dst_input, *outputs)) { in UpdateNode()