Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc2334 llvm::SmallVector<llvm::Value*> input_window_multidim; in EmitScatter() local
2338 input_window_multidim.reserve(rank); in EmitScatter()
2343 input_window_multidim.push_back(index.GetConstantWithIndexType(0)); in EmitScatter()
2347 input_window_multidim.push_back( in EmitScatter()
2352 DCHECK_EQ(input_window_multidim.size(), desc.operand_shape.rank()); in EmitScatter()
2392 Add(input_window_multidim[operand_dim], casted_scatter_index); in EmitScatter()
2393 input_window_multidim[operand_dim] = dim_offset; in EmitScatter()
2411 input_window_multidim, desc.output.GetShape(), index.GetType()); in EmitScatter()