Home
last modified time | relevance | path

Searched refs:output_window_dim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1888 int64 output_window_dim = dim_numbers.offset_dims(operand_index_dim++); in EmitElementalGather() local
1889 operand_to_output_dim[i] = output_window_dim; in EmitElementalGather()
1890 operand_multi_index.push_back(index[output_window_dim]); in EmitElementalGather()
Dhlo_instructions.cc2345 for (int64 output_window_dim : offset_dims) { in MakeGatherDimNumbers() local
2346 gather_dim_numbers.add_offset_dims(output_window_dim); in MakeGatherDimNumbers()