Home
last modified time | relevance | path

Searched refs:output_dimension (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc122 int64 output_dimension = in MakePadding() local
125 std::max<int64>((output_dimension - 1) * window_stride + in MakePadding()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc484 int64 output_dimension = broadcast->dimensions()[operand_dimension]; in HandleBroadcast() local
485 TF_RET_CHECK((output_dimension < broadcast->shape().rank()) && in HandleBroadcast()
486 output_dimension >= 0 && in HandleBroadcast()
487 (broadcast->shape().dimensions(output_dimension) == in HandleBroadcast()