Searched refs:init_value_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_reduce_op.cc | 57 const TensorShape init_value_shape = context->InputShape(n_); in Compile() local
61 OP_REQUIRES(context, TensorShapeUtils::IsScalar(init_value_shape), in Compile()
63 init_value_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2188 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape() argument
2190 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferReduceWindowShape()
2193 return InferReduceWindowShape(operand_shape, init_value_shape, window); in InferReduceWindowShape()
2235 const Shape& operand_shape, const Shape& init_value_shape, in InferReduceWindowShape() argument
2239 init_value_shape.element_type(), in InferReduceWindowShape()
2246 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()
3566 const Shape init_value_shape = in InferScatterShape() local
3568 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, {&init_value_shape}, in InferScatterShape()
|
| D | shape_inference.h | 186 const Shape& init_value_shape, const ProgramShape& scatter_shape);
|
| D | shape_inference_test.cc | 327 Shape init_value_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
332 matrix_shape, init_value_shape, window, to_apply); in TEST_F()
|