Searched refs:scatter_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 2210 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
2219 EXPECT_TRUE(ShapeUtil::Equal(scatter_shape, matrix_64_48_)) in TEST_F()
2220 << ShapeUtil::HumanString(scatter_shape); in TEST_F()
2224 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
2233 EXPECT_TRUE(ShapeUtil::Equal(scatter_shape, matrix_64_48_)) in TEST_F()
2234 << ShapeUtil::HumanString(scatter_shape); in TEST_F()
2238 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
2247 EXPECT_TRUE(ShapeUtil::Equal(scatter_shape, matrix_64_48_)) in TEST_F()
2248 << ShapeUtil::HumanString(scatter_shape); in TEST_F()
2252 TF_ASSERT_OK_AND_ASSIGN(Shape scatter_shape, in TEST_F()
[all …]
|
| D | shape_inference.h | 167 const Shape& init_value_shape, const ProgramShape& scatter_shape);
|
| D | shape_inference.cc | 2127 const Shape& init_value_shape, const ProgramShape& scatter_shape) { in InferSelectAndScatterShape() argument
2163 TF_RETURN_IF_ERROR(VerifyReducerShape(scatter_shape, {&init_value_shape}, in InferSelectAndScatterShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2275 TF_ASSIGN_OR_RETURN(const ProgramShape& scatter_shape, in SelectAndScatterWithGeneralPadding()
2283 source_shape, init_shape, scatter_shape)); in SelectAndScatterWithGeneralPadding()
|