Searched refs:use_linear_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 236 bool use_linear_index = true) const;
253 bool use_linear_index = true) const;
260 bool use_linear_index = true) const;
|
| D | ir_array.cc | 444 bool use_linear_index) const { in EmitArrayElementAddress()
454 if (use_linear_index && index.LinearValidOnShape(shape_)) { in EmitArrayElementAddress()
504 bool use_linear_index) const { in EmitReadArrayElement()
506 EmitArrayElementAddress(index, b, name, use_linear_index); in EmitReadArrayElement()
514 bool use_linear_index) const { in EmitWriteArrayElement()
516 EmitArrayElementAddress(index, b, "", use_linear_index); in EmitWriteArrayElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 367 const llvm_ir::IrArray::Index& index, bool use_linear_index,
|
| D | ir_emitter_unnested.cc | 2119 const IrArray::Index& index, bool use_linear_index, in EmitExtraOutputsForReduce() argument
2132 index, extra_output_ir_values[i], &b_, use_linear_index); in EmitExtraOutputsForReduce()
|