Searched refs:new_result_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_pad_for_tensor_cores.cc | 76 const Shape& new_result_shape) { in PadConv() argument
94 {new_result_shape, ShapeUtil::MakeShape(U8, {0})}); in PadConv()
100 if (!ShapeUtil::Equal(result_shape, new_result_shape)) { in PadConv()
107 HloInstruction::CreateGetTupleElement(new_result_shape, new_conv, 0)); in PadConv()
141 Shape new_result_shape = conv->shape().tuple_shapes(0); in PadForTensorCores() local
153 &new_result_shape); in PadForTensorCores()
155 return std::make_tuple(&new_result_shape, &new_rhs_shape, in PadForTensorCores()
158 return std::make_tuple(&new_lhs_shape, &new_result_shape, in PadForTensorCores()
203 !check_size_increase(result_shape, new_result_shape)) { in PadForTensorCores()
216 PadConv(conv, new_lhs_shape, new_rhs_shape, new_result_shape)); in PadForTensorCores()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 588 Shape new_result_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in ReshapeToAddDegenerateDims() local
596 InlinedVectorToVector(new_output_dims), new_result_shape); in ReshapeToAddDegenerateDims()
|