Searched refs:PadArray2D (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pad_test.cc | 336 auto expected = ReferenceUtil::PadArray2D(*ones, padding_config, 0.0f); in XLA_TEST_P()
358 auto expected = ReferenceUtil::PadArray2D(*operand, padding_config, 3.14f); in XLA_TEST_P()
384 auto expected = ReferenceUtil::PadArray2D(*operand, padding_config, 2.718f); in XLA_TEST_P()
412 auto expected = ReferenceUtil::PadArray2D(*operand, padding_config, 2.718f); in XLA_TEST_P()
440 auto expected = ReferenceUtil::PadArray2D(*operand, padding_config, 2.718f); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 502 static std::unique_ptr<Array2D<NativeT>> PadArray2D( in PadArray2D() function
|