Searched refs:init_value_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_reduce_op.cc | 45 const TensorShape init_value_shape = context->InputShape("init_value"); in Compile() local
49 OP_REQUIRES(context, TensorShapeUtils::IsScalar(init_value_shape), in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2113 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape() argument
2116 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferReduceWindowShape()
2120 init_value_shape.element_type(), in InferReduceWindowShape()
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()
3191 const Shape init_value_shape = in InferScatterShape() local
3193 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferScatterShape()
|
| D | shape_inference.h | 167 const Shape& init_value_shape, const ProgramShape& scatter_shape);
|
| D | shape_inference_test.cc | 315 Shape init_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
320 matrix_shape, init_value_shape, window, to_apply); in TEST_F()
|