Searched refs:ConvWithGeneralDimensions (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 439 ConvWithGeneralDimensions(input, filter, {1, 1, 1}, Padding::kValid, dnums); in XLA_TEST_F()
502 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums); in RunTest()
561 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
627 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
697 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
761 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
830 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
899 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
970 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in RunTest()
1045 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()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 442 XlaOp ConvWithGeneralDimensions(
832 friend XlaOp ConvWithGeneralDimensions(
1411 XlaOp ConvWithGeneralDimensions(
|
| D | xla_builder.cc | 1215 return ConvWithGeneralDimensions( in Conv()
1231 XlaOp XlaBuilder::ConvWithGeneralDimensions( in ConvWithGeneralDimensions() function in xla::XlaBuilder
3054 XlaOp ConvWithGeneralDimensions( in ConvWithGeneralDimensions() function
3059 return lhs.builder()->ConvWithGeneralDimensions( in ConvWithGeneralDimensions()
|
| D | xla_builder_test.cc | 708 ConvWithGeneralDimensions(input, filter, {1, 1}, Padding::kValid, dnums, in TEST_F()
|