Searched refs:output_dimension (Results 1 – 3 of 3) sorted by relevance
122 int64 output_dimension = in MakePadding() local125 std::max<int64>((output_dimension - 1) * window_stride + in MakePadding()
705 int64 output_dimension = broadcast->dimensions()[operand_dimension]; in HandleBroadcast() local706 TF_RET_CHECK((output_dimension < broadcast->shape().rank()) && in HandleBroadcast()707 output_dimension >= 0 && in HandleBroadcast()708 (broadcast->shape().dimensions(output_dimension) == in HandleBroadcast()
1205 int64 output_dimension = input_dynamic_dimension; in HandleGather() local1209 output_dimension--; in HandleGather()1212 parent_->SetDynamicSize(hlo, {}, output_dimension, dynamic_size); in HandleGather()