Searched refs:operand_arrays (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 160 Status EmitDynamicUpdateSliceInPlace(absl::Span<const IrArray> operand_arrays, in EmitDynamicUpdateSliceInPlace() argument
168 IrArray update_array = operand_arrays[1]; in EmitDynamicUpdateSliceInPlace()
169 IrArray start_indices_array = operand_arrays[2]; in EmitDynamicUpdateSliceInPlace()
174 return operand_arrays[2 + index].EmitReadArrayElement( in EmitDynamicUpdateSliceInPlace()
|
| D | dynamic_update_slice_util.h | 66 Status EmitDynamicUpdateSliceInPlace(absl::Span<const IrArray> operand_arrays,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 504 absl::Span<const llvm_ir::IrArray> operand_arrays,
571 absl::Span<const llvm_ir::IrArray> operand_arrays,
|
| D | ir_emitter_unnested.cc | 1779 auto operand_arrays = in EmitLoopFusionFromMlir() local
1800 auto ir_array = operand_arrays[i]; in EmitLoopFusionFromMlir()
4145 absl::Span<const llvm_ir::IrArray> operand_arrays, in EmitTileElementForFusion() argument
4156 for (int i = 0; i < operand_arrays.size(); i++) { in EmitTileElementForFusion()
4173 auto array = operand_arrays[i]; in EmitTileElementForFusion()
4818 absl::Span<const llvm_ir::IrArray> operand_arrays, in EmitHlo021Tile() argument
4863 param_arrays.push_back(operand_arrays[id]); in EmitHlo021Tile()
4889 EmitTileElementForFusion(fusion, operand_arrays, output_arrays, index, in EmitHlo021Tile()
|