Home
last modified time | relevance | path

Searched defs:mask_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc173 Shape mask_shape = in GetExpandedFilterMask() local
Ddynamic_padder.cc294 const Shape mask_shape = in PadWithScalar() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc991 auto mask_shape = ShapeUtil::ChangeElementType(index_shape, PRED); in ExchangeHaloAndGetValidData() local
Dspmd_partitioner.cc589 auto mask_shape = ShapeUtil::ChangeElementType(index_shape, PRED); in PadWithValueHlo() local