Searched defs:arg_shape (Results 1 – 7 of 7) sorted by relevance
112 xla::Shape arg_shape = shape_status.ConsumeValueOrDie(); in Compile() local
72 const InferredShape& arg_shape = it->second; in PropagateShapes() local
235 for (const auto& arg_shape : argument_shapes) { in Compile() local
357 const Shape* arg_shape = nullptr; in InferConcatOpShape() local1080 const Shape* arg_shape = arg_shapes[0]; in InferMapShape() local2852 const Shape& arg_shape = *arg_shapes[i]; in InferCallShape() local
802 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local813 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local824 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
245 const auto& arg_shape = arg_literals[i]->shape(); in Evaluate() local
1717 const Shape& arg_shape = arg_literals[0]->shape(); in HandleReduce() local