Home
last modified time | relevance | path

Searched defs:use_fast_memory (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler_test.cc310 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
356 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
411 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
1172 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
1215 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
1351 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
1424 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
H A Dlayout_util.cc38 const std::optional<xla::HloSharding>& sharding, bool use_fast_memory, in RewriteLayoutWithShardedShape()
H A Dxla_helpers.cc139 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in IdentityShapeRepresentationFn()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dlayout_util.cc22 const std::optional<xla::HloSharding>& sharding, bool use_fast_memory, in RewriteLayoutWithShardedShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_tpu_device.cc57 const TensorShape& shape, DataType type, bool use_fast_memory, in TpuShapeRepresentation()