Home
last modified time | relevance | path

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

/external/llvm-project/mlir/integration_test/Dialect/Linalg/CPU/
Dtest-subtensor-insert-multiple-uses.mlir9 %insert_val = constant dense<20.0> : tensor<1xf32>
15 …%inserted_at_position_0 = subtensor_insert %insert_val into %const[0][1][1] : tensor<1xf32> into t…
16 …%inserted_at_position_1 = subtensor_insert %insert_val into %const[1][1][1] : tensor<1xf32> into t…
Dtest-subtensor-insert.mlir9 %insert_val = constant dense<20.0> : tensor<1xf32>
10 %inserted = subtensor_insert %insert_val into %const[0][1][1] : tensor<1xf32> into tensor<2xf32>
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc533 llvm::Value* insert_val; in EmitFullWarpShuffleDown() local
535 insert_val = EmitNVPTXShflDown(builder->CreateExtractElement(x, i), in EmitFullWarpShuffleDown()
538 insert_val = EmitAMDGPUShflDown(builder->CreateExtractElement(x, i), in EmitFullWarpShuffleDown()
543 x = builder->CreateInsertElement(x, insert_val, i); in EmitFullWarpShuffleDown()