Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc131 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in IdentityShapeRepresentationFn() argument
140 const absl::optional<xla::HloSharding>& sharding, bool use_fast_memory, in RewriteLayoutWithShardedShape() argument
172 use_fast_memory)); in RewriteLayoutWithShardedShape()
Dxla_helpers.h87 const absl::optional<xla::HloSharding>& sharding, bool use_fast_memory,
Dxla_compiler_test.cc308 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F() argument
351 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F() argument
355 if (use_fast_memory) { in TEST_F()
404 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F() argument
1100 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F() argument
1141 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F() argument
1275 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F() argument
1346 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc55 bool use_fast_memory) { in TpuShapeRepresentation() argument
63 &se_shape.value, type, use_fast_memory, &tpu_shape.value, in TpuShapeRepresentation()
Dxla_device_context.cc98 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in XlaDeviceContext() argument
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor_c_api.h338 XLA_Shape* serialized_xla_shape, int data_type, bool use_fast_memory,