Searched refs:update_dim_size (Results 1 – 4 of 4) sorted by relevance
109 llvm::Value* update_dim_size = llvm::ConstantInt::get( in EmitDynamicUpdateSliceInPlaceImpl() local114 llvm::Value* max_bound = b->CreateSub(output_dim_size, update_dim_size); in EmitDynamicUpdateSliceInPlaceImpl()
2676 const int64 update_dim_size = update_shape.dimensions(dim); in InferDynamicUpdateSliceShape() local2677 if (update_dim_size < 0) { in InferDynamicUpdateSliceShape()2680 update_dim_size); in InferDynamicUpdateSliceShape()2682 if (update_dim_size > input_dim_size) { in InferDynamicUpdateSliceShape()2685 update_dim_size, input_dim_size); in InferDynamicUpdateSliceShape()2687 VLOG(2) << StrFormat("update_sizes[%d] = %d", dim, update_dim_size); in InferDynamicUpdateSliceShape()
2384 int64 update_dim_size =2390 operand_shape.dimensions(i) - update_dim_size)) {
2052 llvm::Value* update_dim_size = in EmitElementalDynamicUpdateSlice() local2067 Add(slice_start_multi_index[i], update_dim_size); in EmitElementalDynamicUpdateSlice()