Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc50 std::vector<llvm::Value*> start_multi_index(rank); in EmitDynamicUpdateSliceInPlaceImpl() local
52 TF_ASSIGN_OR_RETURN(start_multi_index[i], start_indices_generator(i)); in EmitDynamicUpdateSliceInPlaceImpl()
54 start_multi_index[i]->getType(), output_shape.dimensions(i)); in EmitDynamicUpdateSliceInPlaceImpl()
56 start_multi_index[i]->getType(), update_shape.dimensions(i)); in EmitDynamicUpdateSliceInPlaceImpl()
62 llvm::ConstantInt::get(start_multi_index[i]->getType(), 0); in EmitDynamicUpdateSliceInPlaceImpl()
63 start_multi_index[i] = in EmitDynamicUpdateSliceInPlaceImpl()
66 zero, start_multi_index[i]), in EmitDynamicUpdateSliceInPlaceImpl()
67 zero, start_multi_index[i]); in EmitDynamicUpdateSliceInPlaceImpl()
69 start_multi_index[i] = in EmitDynamicUpdateSliceInPlaceImpl()
72 max_bound, start_multi_index[i]), in EmitDynamicUpdateSliceInPlaceImpl()
[all …]