Searched defs:shape_0 (Results 1 – 5 of 5) sorted by relevance
41 std::vector<int> shape_0; in IsBroadcastable() local
95 const BHWC shape_0{input_shape.b, 36, tiles_x * tiles_y, input_shape.c}; in WinogradFromNode() local
252 const int64 shape_0 = params.shape(0); in SetupGraph() local
447 bool ShapeVerifier::HasCompatibleElementTypes(const Shape& shape_0, in HasCompatibleElementTypes()460 const Shape& shape_0 = instruction->operand(0)->shape(); in HandleRng() local
1269 auto shape_0 = ops::Shape(s.WithOpName("s"), conv_0); in TEST_F() local