Searched refs:max_shape (Results 1 – 4 of 4) sorted by relevance
31 const TensorShape max_shape = ctx->InputShape(2); in Compile() local43 " clip_value_max shape: ", max_shape.DebugString()); in Compile()50 if (shape != max_shape) { in Compile()51 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(max_shape), shape_error()); in Compile()
115 auto max_shape = max_signed->shape(); in ExpandInstruction() local116 max_shape.set_element_type(unsigned_type); in ExpandInstruction()118 HloInstruction::CreateConvert(max_shape, max_signed)); in ExpandInstruction()120 unsigned_shape, max_unsigned, max_shape.dimensions())); in ExpandInstruction()
408 max_shape = math_ops.reduce_max(413 max_shape[:axis], shape[-1:]415 if axis == -1 else max_shape[axis + 1:]
338 ShapeHandle max_shape; in __anon30b5031d0302() local339 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &max_shape)); in __anon30b5031d0302()