Lines Matching refs:ShapeUtil
46 return ShapeUtil::ByteSizeOf(shape, kPointerSize);
66 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
77 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
78 auto y = builder.Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
88 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
99 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
100 auto y = builder.Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
110 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
111 auto y = builder.Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
149 auto lhs = builder.Parameter(0, ShapeUtil::MakeShape(F32, {10, 5}), "lhs");
150 auto rhs = builder.Parameter(1, ShapeUtil::MakeShape(F32, {5, 30}), "rhs");
171 auto input = builder.Parameter(0, ShapeUtil::MakeShape(F32, {10}), "in");
189 0, ShapeUtil::MakeShape(F32, {/*p_dim=*/1, /*z_dim=*/1, /*y_dim=*/10,
193 1, ShapeUtil::MakeShape(F32, {/*p_dim=*/1, /*z_dim=*/1, /*y_dim=*/3,
216 builder.Parameter(0, ShapeUtil::MakeShape(F32, {10, 20}), "input");
234 builder.Parameter(0, ShapeUtil::MakeShape(F32, {10, 20}), "input");
251 builder.Parameter(0, ShapeUtil::MakeShape(F32, {10, 20}), "input");
253 builder.Parameter(1, ShapeUtil::MakeShape(F32, {2, 4}), "source");
283 builder.Parameter(0, ShapeUtil::MakeShape(F32, {10, 5}), "input");
285 builder.Parameter(1, ShapeUtil::MakeShape(F32, {5, 20}), "weight");
286 auto bias = builder.Parameter(2, ShapeUtil::MakeShape(F32, {20}), "bias");
307 auto lhs = builder.Parameter(0, ShapeUtil::MakeShape(F32, {64, 64, 1, 1}),
309 auto rhs = builder.Parameter(1, ShapeUtil::MakeShape(F32, {64, 64, 1, 1}),
321 builder.Parameter(0, ShapeUtil::MakeShape(F32, {64, 64}), "input");
323 builder.Parameter(1, ShapeUtil::MakeShape(F32, {64, 64}), "weights");
339 Shape r2f32 = ShapeUtil::MakeShape(F32, {2, 2});
397 ShapeUtil::MakeShape(F32, {2, 3, 4, 5});
429 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {123}), "x");
430 auto y = builder.Parameter(1, ShapeUtil::MakeShape(F32, {42}), "y");
