Searched refs:scale_offset_shape (Results 1 – 1 of 1) sorted by relevance
1450 const TensorShape& scale_offset_shape = scale.shape(); in ComputeWithReservedSpace() local1452 OP_REQUIRES_OK(context, context->allocate_output(1, scale_offset_shape, in ComputeWithReservedSpace()1455 OP_REQUIRES_OK(context, context->allocate_output(2, scale_offset_shape, in ComputeWithReservedSpace()