Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dscatter_expander.cc191 std::vector<int64_t> max_valid_index(operand_dims.size()); in CheckIndexValidity() local
193 max_valid_index[i] = operand_dims[i] - window_sizes[i]; in CheckIndexValidity()
198 max_valid_index)); in CheckIndexValidity()
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dcuda_kernel.py207 def max_valid_index(self, node: IRNode, default=-1): member in CUDATemplateKernel