Home
last modified time | relevance | path

Searched refs:scatter_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc2912 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
2921 EXPECT_TRUE(ShapeUtil::Equal(scatter_shape, matrix_64_48_)) in TEST_F()
2922 << ShapeUtil::HumanString(scatter_shape); in TEST_F()
2926 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
2935 EXPECT_TRUE(ShapeUtil::Equal(scatter_shape, matrix_64_48_)) in TEST_F()
2936 << ShapeUtil::HumanString(scatter_shape); in TEST_F()
2940 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
2949 EXPECT_TRUE(ShapeUtil::Equal(scatter_shape, matrix_64_48_)) in TEST_F()
2950 << ShapeUtil::HumanString(scatter_shape); in TEST_F()
2954 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
[all …]
Dshape_inference.h186 const Shape& init_value_shape, const ProgramShape& scatter_shape);
Dshape_inference.cc2246 const Shape& init_value_shape, const ProgramShape& scatter_shape) { in InferSelectAndScatterShape() argument
2282 TF_RETURN_IF_ERROR(VerifyReducerShape(scatter_shape, {&init_value_shape}, in InferSelectAndScatterShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3084 TF_ASSIGN_OR_RETURN(const ProgramShape& scatter_shape, in SelectAndScatterInternal()
3093 *source_shape, *init_shape, scatter_shape)); in SelectAndScatterInternal()