Home
last modified time | relevance | path

Searched refs:array4d (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dround_trip_transfer_test.cc136 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()
DBUILD93 "//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 …]
Darray_elementwise_ops_test.cc2004 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/
DBUILD373 ":array4d",
394 ":array4d",
418 ":array4d",
546 name = "array4d",
547 hdrs = ["array4d.h"],
563 ":array4d",
750 ":array4d",
773 ":array4d",
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD135 "//tensorflow/compiler/xla:array4d",
211 "//tensorflow/compiler/xla:array4d",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD424 "//tensorflow/compiler/xla:array4d",