Home
last modified time | relevance | path

Searched refs:ConvGeneralDilated (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc320 xla::ConvGeneralDilated(input_data, kernel, dims.stride, in ResizeUsingDilationAndConvolution()
330 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolution()
339 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolution()
387 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolutionGradOp()
415 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolutionGradOp()
423 output = xla::ConvGeneralDilated( in ResizeUsingDilationAndConvolutionGradOp()
Dxla_conv_op.cc77 xla::XlaOp output = xla::ConvGeneralDilated( in Compile()
Dconv_op_helpers.cc318 return xla::ConvGeneralDilated( in MakeXlaForwardConvOp()
385 return xla::ConvGeneralDilated( in MakeXlaBackpropInputConvOp()
532 filter_backprop = xla::ConvGeneralDilated( in MakeXlaBackpropFilterConvOp()
Dextract_image_patches_op.cc150 xla::ConvGeneralDilated(ctx->Input(0), filter, window_strides, padding, in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_variants_test.cc640 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 …]
Dconvolution_test.cc281 ConvGeneralDilated( in RunTest()
317 ConvGeneralDilated( in XLA_TEST_F()
348 ConvGeneralDilated( in XLA_TEST_F()
382 ConvGeneralDilated( in RunTest()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1635 return self.ConvGeneralDilated(
1655 return self.ConvGeneralDilated(
1674 def ConvGeneralDilated(self, lhs, rhs, window_strides, padding, lhs_dilation, member in ComputationBuilder
1729 return self._client.ConvGeneralDilated(
Dlocal_computation_builder.cc586 LocalOp ComputationBuilder::ConvGeneralDilated( in ConvGeneralDilated() function in xla::swig::ComputationBuilder
593 return xla::ConvGeneralDilated(lhs.op(), rhs.op(), window_strides, padding, in ConvGeneralDilated()
Dlocal_computation_builder.h293 LocalOp ConvGeneralDilated(
Dlocal_computation_builder.i339 %unignore xla::swig::ComputationBuilder::ConvGeneralDilated;
Dxla_client_test.py728 c.ConvGeneralDilated(c.Constant(lhs), c.Constant(rhs),
748 c.ConvGeneralDilated(c.Constant(np.transpose(lhs, (0, 2, 3, 1))),
769 c.ConvGeneralDilated(c.Constant(lhs), c.Constant(rhs),
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h409 XlaOp ConvGeneralDilated(const XlaOp& lhs, const XlaOp& rhs,
803 friend XlaOp ConvGeneralDilated(
1337 XlaOp ConvGeneralDilated(const XlaOp& lhs, const XlaOp& rhs,
Dxla_builder.cc1161 return ConvGeneralDilated(lhs, rhs, window_strides, padding, {}, {}, in ConvGeneral()
1166 XlaOp XlaBuilder::ConvGeneralDilated( in ConvGeneralDilated() function in xla::XlaBuilder
3010 XlaOp ConvGeneralDilated(const XlaOp& lhs, const XlaOp& rhs, in ConvGeneralDilated() function
3018 return lhs.builder()->ConvGeneralDilated( in ConvGeneralDilated()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc581 ConvGeneralDilated(input, kernel, /*window_strides=*/{1, 1}, in TEST_F()