Home
last modified time | relevance | path

Searched defs:new_result_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores.cc76 const Shape& new_result_shape) { in PadConv()
141 Shape new_result_shape = conv->shape().tuple_shapes(0); in PadForTensorCores() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc588 Shape new_result_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in ReshapeToAddDegenerateDims() local