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 | 1938 ConvGeneral(input, filter, /*window_strides=*/{1, 1}, in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 402 XlaOp ConvGeneral(const XlaOp& lhs, const XlaOp& rhs, 797 friend XlaOp ConvGeneral(const XlaOp& lhs, const XlaOp& rhs, 1328 XlaOp ConvGeneral(const XlaOp& lhs, const XlaOp& rhs,
|
D | xla_builder.cc | 1114 return ConvGeneral(lhs, rhs, window_strides, padding, in ConvWithGeneralPadding() 1147 return ConvGeneral(lhs, rhs, window_strides, in ConvWithGeneralDimensions() 1155 XlaOp XlaBuilder::ConvGeneral( in ConvGeneral() function in xla::XlaBuilder 2999 XlaOp ConvGeneral(const XlaOp& lhs, const XlaOp& rhs, in ConvGeneral() function 3005 return lhs.builder()->ConvGeneral(lhs, rhs, window_strides, padding, in ConvGeneral()
|