Home
last modified time | relevance | path

Searched defs:temporary_memory (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dtemporary_memory_manager.h139 port::StatusOr<std::unique_ptr<TemporaryDeviceMemoryBase>> temporary_memory = in AllocateArray() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.h55 Tensor temporary_memory; in AllocateBytes() local
H A Dfused_batch_norm_op.cc686 Tensor temporary_memory; in AllocateBytes() local
745 Tensor* temporary_memory = nullptr; in AllocateBytes() local
H A Dfft_ops.cc381 Tensor temporary_memory; in AllocateBytes() local
H A Dcudnn_rnn_ops.cc374 Tensor temporary_memory; in AllocateBytes() local
423 Tensor* temporary_memory = nullptr; in AllocateBytes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.h75 int64_t temporary_memory = 0; member
H A Dcost_estimator.h150 int64_t temporary_memory; member