Home
last modified time | relevance | path

Searched defs:shape_builder (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dnccl_rewrite.cc208 NodeBuilder shape_builder(shape_name, "Shape"); in ReplaceBroadcast() local
222 NodeBuilder shape_builder(strings::StrCat(shape_name, recv_index), in ReplaceBroadcast() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
H A Dserialization.cc238 data::Uint3Builder shape_builder(*builder); in operator ()() local
245 data::Uint2Builder shape_builder(*builder); in operator ()() local
251 data::Uint1Builder shape_builder(*builder); in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dserialization_base.cc396 data::BHWDCBuilder shape_builder(*builder); in Encode() local