| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 220 int64 output_batch_dimension = dim_numbers.output_batch_dimension(); in HandleBatchGroupCount() local
283 if (output_batch_dimension > output_feature_dimension) { in HandleBatchGroupCount()
284 dim_numbers.set_output_batch_dimension(output_batch_dimension + 1); in HandleBatchGroupCount()
326 GetExpandedFilterMask(convolution->shape(), output_batch_dimension, in HandleBatchGroupCount()
329 convolution->shape(), batch_group_count, output_batch_dimension); in HandleBatchGroupCount()
331 VLOG(2) << "output_batch_dimension " << output_batch_dimension; in HandleBatchGroupCount()
353 reduce_window_shape.set_dimensions(output_batch_dimension, 1); in HandleBatchGroupCount()
389 if (i == output_batch_dimension) { in HandleBatchGroupCount()
552 if (dim_numbers.output_batch_dimension() > in HandleConvolution()
555 dim_numbers.output_batch_dimension() + 1); in HandleConvolution()
[all …]
|
| D | space_to_batch_converter_test.cc | 62 .output_batch_dimension(); in TEST_F()
111 .output_batch_dimension(); in TEST_F()
145 .output_batch_dimension(); in TEST_F()
|
| D | convolution_4d_expander.cc | 137 removed_output_dimensions, new_dim_nums.output_batch_dimension())); in ExpandInstruction()
|
| D | dot_as_convolution_util.cc | 59 conv_dims.output_batch_dimension(), -1}); in ParseConvolutionDimsInfo()
|
| D | space_to_batch_converter.cc | 1991 dim_map[permuted_conv_dims_numbers.output_batch_dimension()] = dim_count; in PropagateOnConv()
2027 std::make_pair(original_conv_dims.output_batch_dimension(), in PropagateOnConv()
2553 std::make_pair(original_conv_dims.output_batch_dimension(), in PropagateOnBackpropFilterConv()
2754 dim_map[dim_numbers.output_batch_dimension()] = dim_count; in PerformSpaceToBatchOnConvolution()
2796 const int64 output_batch_dim = new_dim_numbers.output_batch_dimension(); in PerformSpaceToBatchOnConvolution()
2806 dim_numbers.output_batch_dimension(), in PerformSpaceToBatchOnConvolution()
2812 std::make_pair(dim_numbers.output_batch_dimension(), in PerformSpaceToBatchOnConvolution()
|
| D | dynamic_dimension_inference.cc | 571 dimension_numbers.output_batch_dimension(), in HandleConvolution()
699 parent_->SetDynamicSize(hlo, {}, dimension_numbers.output_batch_dimension(), in HandleDynamicConvolutionForward()
|
| D | shape_inference.cc | 1697 output_dnums[0] = dnums.output_batch_dimension(); in InferConvolveShape()
1821 dimensions[dnums.output_batch_dimension()] = input_batch / batch_group_count; in InferConvolveShape()
1832 is_dynamic[dnums.output_batch_dimension()] = true; in InferConvolveShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 43 output_batch_dimension = 0 : i64,
78 output_batch_dimension = 0 : i64,
108 output_batch_dimension = 0 : i64,
140 output_batch_dimension = 0 : i64,
171 output_batch_dimension = 0 : i64,
|
| D | lhlo-legalize-to-linalg.mlir | 908 … i64, kernel_spatial_dimensions = dense<[0, 1]> : tensor<2xi64>, output_batch_dimension = 0 : i64,…
915 … i64, kernel_spatial_dimensions = dense<[0, 1]> : tensor<2xi64>, output_batch_dimension = 0 : i64,…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | layouts_and_names.mlir | 17 output_batch_dimension = 0 : i64,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 106 dnums.output_batch_dimension() == 0 && in PotentiallyImplementedAsEigenConvolution()
|
| D | conv_canonicalization.cc | 103 auto output_batch_dim = dnums.output_batch_dimension(); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 545 self.output_batch_dimension = 0
585 dimension_numbers.output_batch_dimension = 0
598 dimension_numbers.output_batch_dimension = out_spec.index('N')
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_dimension_numbers_test.cc | 109 int64 old_output_batch_dim = dim_nums.output_batch_dimension(); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.td | 48 StructFieldAttr<"output_batch_dimension", I64Attr>,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 99 output_layout.push_back(dnums.output_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
106 output_layout.push_back(dnums.output_batch_dimension()); in StreamExecutorConvLayoutsToXlaLayouts()
|
| D | gpu_conv_rewriter.cc | 187 auto output_batch_dim = conv_dnums.output_batch_dimension(); in MatchBackwardFilter()
516 dnums.set_input_batch_dimension(conv_dnums.output_batch_dimension()); in MatchBackwardInput()
|
| D | gpu_conv_runner.cc | 421 .set_count(output_shape.dimensions(dnums.output_batch_dimension())); in GetGpuConvConfig()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_exporter.cc | 49 input.output_batch_dimension().getValue().getSExtValue()); in ConvertConvDimensionNumbers()
|
| D | attribute_importer.cc | 122 builder->getI64IntegerAttr(dnums.output_batch_dimension()), in ConvertConvDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 290 // CHECK-SAME: output_batch_dimension = 0 : i64
327 // CHECK-SAME: output_batch_dimension = 0 : i64
366 // CHECK-SAME: output_batch_dimension = 0 : i64
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 88 dnums.output_batch_dimension(); in PartitionConvolutionWithBatchGroupCount()
179 dnums.output_batch_dimension(); in PartitionConvolutionWithFeatureGroupCount()
759 dnums.output_batch_dimension(); in PartitionConvolutionTiledOutput()
|
| D | dot_handler.cc | 281 int64 output_batch_dimension = dnums.output_batch_dimension(); in GenNewConvDNums() local
290 if (output_slice_dim <= output_batch_dimension) { in GenNewConvDNums()
291 output_batch_dimension++; in GenNewConvDNums()
318 new_dnums.set_output_batch_dimension(output_batch_dimension); in GenNewConvDNums()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 546 conv->shape().tuple_shapes(0), dim_nums.output_batch_dimension(), in EmitConvolutionForwardAsMlir()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 140 dnums.output_batch_dimension = 0
|