Searched refs:fast_mem (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 183 absl::optional<xla::OpSharding> sharding, bool fast_mem) { in ReshapeWithCorrectRepresentationAndSharding() argument
192 shape_representation_fn, subsharding, fast_mem)); in ReshapeWithCorrectRepresentationAndSharding()
203 shape_representation_fn(shape, dtype, fast_mem)); in ReshapeWithCorrectRepresentationAndSharding()
208 hlo_sharding, fast_mem, shape_representation_fn, &to_shape)); in ReshapeWithCorrectRepresentationAndSharding()
|
| D | xla_argument.h | 95 bool fast_mem = false; member
|
| D | xla_helpers.h | 96 absl::optional<xla::OpSharding> sharding, bool fast_mem);
|
| D | xla_compiler.cc | 335 shape_representation_fn, sharding, arg.fast_mem)); in BuildComputation()
455 " initialized=", initialized, " is_fast_mem=", fast_mem); in HumanString()
880 /*use_fast_memory=*/arg.fast_mem)); in XLAShapeForArgument()
882 arg_sharding, arg.fast_mem, options_.shape_representation_fn, in XLAShapeForArgument()
|
| D | xla_compiler_test.cc | 345 args[0].fast_mem = true; in TEST_F()
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | compile_metadata.proto | 57 bool fast_mem = 7; field
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 203 arg.fast_mem = proto_arg.fast_mem(); in BuildComputationArgumentDescriptions()
|