Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1011 std::vector<llvm::Value*> input_spatial(num_spatial_dims); in EmitElementalConvolution() local
1013 input_spatial[i] = calculate_input_index( in EmitElementalConvolution()
1035 llvm::Value* dim_in_bound = ICmpULT(input_spatial[i], input_bound); in EmitElementalConvolution()
1037 not_in_hole(input_spatial[i], window.dimensions(i).base_dilation()); in EmitElementalConvolution()
1048 input_spatial[i] = in EmitElementalConvolution()
1049 undilate(input_spatial[i], window.dimensions(i).base_dilation()); in EmitElementalConvolution()
1060 input_multi_index[dnums.input_spatial_dimensions(i)] = input_spatial[i]; in EmitElementalConvolution()