Home
last modified time | relevance | path

Searched defs:operand_buffers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc459 absl::Span<se::DeviceMemoryBase> operand_buffers, in GetGpuConvParams()
496 absl::Span<se::DeviceMemoryBase> operand_buffers, in RunGpuConv()
506 absl::Span<se::DeviceMemoryBase> operand_buffers, in RunGpuConv()
Dgpu_conv_algorithm_picker.cc126 absl::Span<se::DeviceMemoryBase> operand_buffers, in GetMIOpenAlgorithms()
405 std::vector<se::DeviceMemoryBase> operand_buffers; in PickBestAlgorithmNoCacheCuda() local
661 std::vector<se::DeviceMemoryBase> operand_buffers; in PickBestAlgorithmNoCacheRocm() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbuffer_reuse_pass.cc89 auto operand_buffers = genericOp.getShapedOperands(); in build() local
305 auto operand_buffers = generic_op.getShapedOperands(); in can_reuse_locally() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc152 PointsToSet::BufferSet* operand_buffers = in OperandBufferForwarded() local