Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2621 std::vector<llvm::Value*> input_window_multidim; in EmitScatter() local
2628 input_window_multidim.push_back(index.GetConstantWithIndexType(0)); in EmitScatter()
2632 input_window_multidim.push_back( in EmitScatter()
2637 DCHECK_EQ(input_window_multidim.size(), desc.operand_shape.rank()); in EmitScatter()
2679 Add(input_window_multidim[operand_dim], casted_scatter_index); in EmitScatter()
2680 input_window_multidim[operand_dim] = dim_offset; in EmitScatter()
2698 input_window_multidim, desc.output.GetShape(), index.GetType()); in EmitScatter()