Home
last modified time | relevance | path

Searched defs:operand_buffer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc48 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoPrecedingAffine() local
119 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc213 for (const BufferValue* operand_buffer : used_buffers[instruction]) { in RunComputation() local
255 for (const BufferValue* operand_buffer : operand_buffers_to_free) { in RunComputation() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc497 auto operand_buffer = GetAllocationSlice(*custom_call->operand(0)); in HandleCustomCall() local
554 auto operand_buffer = GetAllocationSlice(*hlo->operand(1)); in HandleTriangularSolve() local
1037 auto operand_buffer = GetAllocationSlice(*operand); in HandleScatter() local