Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5097 std::vector<int64> params_012; in CheckAndEmitHloWithTile021() local
5116 params_012.push_back(operand_idx); in CheckAndEmitHloWithTile021()
5129 params_012 = FilterInputsForShmemTranspose(fusion_op, params_012); in CheckAndEmitHloWithTile021()
5130 if (params_012.empty()) { in CheckAndEmitHloWithTile021()
5156 for (int64 i = 0; i < params_012.size(); ++i) { in CheckAndEmitHloWithTile021()
5157 const Shape& operand_shape = context.operand_shapes[params_012[i]]; in CheckAndEmitHloWithTile021()
5164 params_012.resize(i); in CheckAndEmitHloWithTile021()
5169 if (params_012.empty()) { in CheckAndEmitHloWithTile021()
5181 *reduced_dims_021, params_012); in CheckAndEmitHloWithTile021()