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 | 510 static std::unique_ptr<Array2D<NativeT>> PadArray2D( in PadArray2D() function
|