Searched refs:shape_builder (Results 1 – 3 of 3) sorted by relevance
208 NodeBuilder shape_builder(shape_name, "Shape"); in ReplaceBroadcast() local209 shape_builder.Input(in_node).Attr("out_type", DT_INT32).Attr("T", dtype); in ReplaceBroadcast()210 TF_RETURN_IF_ERROR(shape_builder.Finalize(graph, &shape_node)); in ReplaceBroadcast()222 NodeBuilder shape_builder(strings::StrCat(shape_name, recv_index), in ReplaceBroadcast() local224 shape_builder.Attr("value", tensor_proto).Attr("dtype", DT_INT32); in ReplaceBroadcast()225 TF_RETURN_IF_ERROR(shape_builder.Finalize(graph, &shape_node)); in ReplaceBroadcast()
234 data::Uint3Builder shape_builder(*builder); in operator ()() local235 shape_builder.add_x(shape.x); in operator ()()236 shape_builder.add_y(shape.y); in operator ()()237 shape_builder.add_z(shape.z); in operator ()()238 return shape_builder.Finish().Union(); in operator ()()241 data::Uint2Builder shape_builder(*builder); in operator ()() local242 shape_builder.add_x(shape.x); in operator ()()243 shape_builder.add_y(shape.y); in operator ()()244 return shape_builder.Finish().Union(); in operator ()()247 data::Uint1Builder shape_builder(*builder); in operator ()() local[all …]
460 data::BHWDCBuilder shape_builder(*builder); in Encode() local461 shape_builder.add_b(desc.shape.b); in Encode()462 shape_builder.add_h(desc.shape.h); in Encode()463 shape_builder.add_w(desc.shape.w); in Encode()464 shape_builder.add_d(desc.shape.d); in Encode()465 shape_builder.add_c(desc.shape.c); in Encode()466 auto shape_fb = shape_builder.Finish(); in Encode()