Searched refs:ConvGeneral (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_variants_test.cc | 704 ConvGeneral( in XLA_TEST_F()
723 ConvGeneral( in XLA_TEST_F()
742 ConvGeneral( in XLA_TEST_F()
998 ConvGeneral(input, filter, {1, 1}, {{2, 1}, {2, 3}}, dnums); in XLA_TEST_F()
1042 ConvGeneral(input, filter, {1, 1}, {{2, 1}, {2, 3}}, dnums); in XLA_TEST_F()
1086 ConvGeneral(input, filter, {1, 1}, {{0, 0}, {0, 0}}, dnums); in XLA_TEST_F()
1127 ConvGeneral(input, filter, {1, 1}, {{0, 0}, {0, 0}}, dnums); in XLA_TEST_F()
|
| D | convolution_test.cc | 1935 ConvGeneral(input, filter, /*window_strides=*/{1, 1}, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 448 XlaOp ConvGeneral(XlaOp lhs, XlaOp rhs,
837 friend XlaOp ConvGeneral(XlaOp lhs, XlaOp rhs,
1419 XlaOp ConvGeneral(XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
|
| D | xla_builder.cc | 1226 return ConvGeneral(lhs, rhs, window_strides, padding, in ConvWithGeneralPadding()
1259 return ConvGeneral(lhs, rhs, window_strides, in ConvWithGeneralDimensions()
1267 XlaOp XlaBuilder::ConvGeneral( in ConvGeneral() function in xla::XlaBuilder
3064 XlaOp ConvGeneral(const XlaOp lhs, const XlaOp rhs, in ConvGeneral() function
3070 return lhs.builder()->ConvGeneral(lhs, rhs, window_strides, padding, in ConvGeneral()
|