Home
last modified time | relevance | path

Searched defs:output_batch_dimension (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddepthwise_convolution_converter.cc118 int64 output_batch_dimension = dim_numbers.output_batch_dimension(); in HandleBackwardFilterBatchGroupConvolution() local
Dconvolution_group_converter.cc220 int64 output_batch_dimension = dim_numbers.output_batch_dimension(); in HandleBatchGroupCount() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto517 int64 output_batch_dimension = 9; field