| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test_1d.cc | 245 ConvGeneralDilated( in RunTest()
281 ConvGeneralDilated( in XLA_TEST_F()
312 ConvGeneralDilated( in XLA_TEST_F()
346 ConvGeneralDilated( in RunTest()
|
| D | convolution_variants_test.cc | 640 ConvGeneralDilated( in XLA_TEST_F()
659 ConvGeneralDilated( in XLA_TEST_F()
682 ConvGeneralDilated( in XLA_TEST_F()
761 ConvGeneralDilated( in XLA_TEST_F()
786 ConvGeneralDilated( in XLA_TEST_F()
1175 ConvGeneralDilated(gradients, mirrored_weights, in XLA_TEST_F()
1237 ConvGeneralDilated(activations, gradients, in XLA_TEST_F()
1263 ConvGeneralDilated(activations, gradients, in XLA_TEST_F()
1290 ConvGeneralDilated(activations, gradients, in XLA_TEST_F()
1322 ConvGeneralDilated(activations, gradients, in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 337 xla::ConvGeneralDilated(input_data, kernel, dims.stride, in ResizeUsingDilationAndConvolution()
349 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolution()
361 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolution()
410 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolutionGradOp()
438 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolutionGradOp()
446 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolutionGradOp()
|
| D | xla_conv_op.cc | 77 xla::XlaOp output = xla::ConvGeneralDilated( in Compile()
|
| D | conv_op_helpers.cc | 286 return xla::ConvGeneralDilated( in MakeXlaForwardConvOp()
384 return xla::ConvGeneralDilated(out_backprop, filter, /*window_strides=*/ones, in MakeXlaBackpropInputConvOp()
536 filter_backprop = xla::ConvGeneralDilated( in MakeXlaBackpropFilterConvOp()
|
| D | extract_image_patches_op.cc | 156 xla::ConvGeneralDilated(ctx->Input(0), filter, window_strides, padding, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 332 ConvGeneralDilated(input, kernel, {kLarge - 1}, {{0, 0}}, {kLarge}, {1}, in TEST_F()
355 ConvGeneralDilated(input, kernel, {1}, {{kLarge - 1, kLarge - 1}}, {1}, {1}, in TEST_F()
892 ConvGeneralDilated(input, kernel, /*window_strides=*/{1, 1}, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 112 ops.def("ConvGeneralDilated", &ConvGeneralDilated, py::arg("lhs"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 761 ops.ConvGeneralDilated(
786 ops.ConvGeneralDilated(
815 ops.ConvGeneralDilated(
836 ops.ConvGeneralDilated(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 562 XlaOp ConvGeneralDilated(
1194 friend XlaOp ConvGeneralDilated(
1905 XlaOp ConvGeneralDilated(
|
| D | xla_builder.cc | 1442 return ConvGeneralDilated(lhs, rhs, window_strides, padding, {}, {}, in ConvGeneral()
1448 XlaOp XlaBuilder::ConvGeneralDilated( in ConvGeneralDilated() function in xla::XlaBuilder
4356 XlaOp ConvGeneralDilated(const XlaOp lhs, const XlaOp rhs, in ConvGeneralDilated() function
4365 return lhs.builder()->ConvGeneralDilated( in ConvGeneralDilated()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 720 xla::XlaOp xla_result = xla::ConvGeneralDilated( in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17657 let summary = "Wraps the XLA ConvGeneralDilated operator, documented at";
|