Searched refs:update_dim_size (Results 1 – 4 of 4) sorted by relevance
55 llvm::Value* update_dim_size = llvm::ConstantInt::get( in EmitDynamicUpdateSliceInPlaceImpl() local60 llvm::Value* max_bound = b->CreateSub(output_dim_size, update_dim_size); in EmitDynamicUpdateSliceInPlaceImpl()
2468 const int64 update_dim_size = update_shape.dimensions(dim); in InferDynamicUpdateSliceShape() local2469 if (update_dim_size < 0) { in InferDynamicUpdateSliceShape()2472 update_dim_size); in InferDynamicUpdateSliceShape()2474 if (update_dim_size > input_dim_size) { in InferDynamicUpdateSliceShape()2477 update_dim_size, input_dim_size); in InferDynamicUpdateSliceShape()2479 VLOG(2) << StrFormat("update_sizes[%d] = %d", dim, update_dim_size); in InferDynamicUpdateSliceShape()
1993 llvm::Value* update_dim_size = in EmitElementalDynamicUpdateSlice() local2008 Add(slice_start_multi_index[i], update_dim_size); in EmitElementalDynamicUpdateSlice()
2363 int64 update_dim_size = in HandleScatter() local2369 operand_shape.dimensions(i) - update_dim_size)) { in HandleScatter()