Searched refs:ConvWithGeneralDimensions (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 271 ConvWithGeneralDimensions(input, filter, {1, 1, 1}, Padding::kValid, dnums); in XLA_TEST_F()
334 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums); in RunTest()
393 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
459 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
529 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
593 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
662 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
731 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
802 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
877 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
[all …]
|
| D | convolution_dimension_numbers_test.cc | 120 ConvWithGeneralDimensions(input, conv1, {1, 1}, Padding::kValid, dim_nums); in XLA_TEST_F()
|
| D | convolution_test_1d.cc | 104 ConvWithGeneralDimensions(input, filter, {1}, Padding::kValid, dnums); in TestImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 796 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in TEST_F()
1122 ConvWithGeneralDimensions(p0, p1, {1, 1}, Padding::kValid, dnums, in TEST_F()
|
| D | xla_builder.h | 547 XlaOp ConvWithGeneralDimensions(
1141 friend XlaOp ConvWithGeneralDimensions(
1886 XlaOp ConvWithGeneralDimensions(
|
| D | xla_builder.cc | 1378 return ConvWithGeneralDimensions( in Conv()
1397 XlaOp XlaBuilder::ConvWithGeneralDimensions( in ConvWithGeneralDimensions() function in xla::XlaBuilder
4333 XlaOp ConvWithGeneralDimensions( in ConvWithGeneralDimensions() function
4339 return lhs.builder()->ConvWithGeneralDimensions( in ConvWithGeneralDimensions()
|