Searched refs:array4d (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | round_trip_transfer_test.cc | 136 Array4D<float> array4d(2, 2, 256, 256); in TEST_F() local 137 array4d.FillWithMultiples(1.0f); in TEST_F() 138 RoundTripTest(LiteralUtil::CreateR4FromArray4D<float>(array4d)); in TEST_F()
|
D | BUILD | 93 "//tensorflow/compiler/xla:array4d", 175 "//tensorflow/compiler/xla:array4d", 658 "//tensorflow/compiler/xla:array4d", 834 "//tensorflow/compiler/xla:array4d", 853 "//tensorflow/compiler/xla:array4d", 908 "//tensorflow/compiler/xla:array4d", 930 "//tensorflow/compiler/xla:array4d", 961 "//tensorflow/compiler/xla:array4d", 995 "//tensorflow/compiler/xla:array4d", 1157 "//tensorflow/compiler/xla:array4d", [all …]
|
D | array_elementwise_ops_test.cc | 2004 auto array4d = ConstantR4FromArray4D<float>( in XLA_TEST_F() local 2007 Min(array2d, array4d, /*broadcast_dimensions=*/{1, 3}); in XLA_TEST_F() 2020 auto array4d = ConstantR4FromArray4D<float>(&builder, arg); in XLA_TEST_F() local 2021 Min(array2d, array4d, /*broadcast_dimensions=*/{1, 3}); in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | BUILD | 373 ":array4d", 394 ":array4d", 418 ":array4d", 546 name = "array4d", 547 hdrs = ["array4d.h"], 563 ":array4d", 750 ":array4d", 773 ":array4d",
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | BUILD | 135 "//tensorflow/compiler/xla:array4d", 211 "//tensorflow/compiler/xla:array4d",
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | BUILD | 424 "//tensorflow/compiler/xla:array4d",
|