Searched defs:input_tensor_dims (Results 1 – 5 of 5) sorted by relevance
21 api::utils::ivec4& input_tensor_dims) { in set_cumsum_kernel_params()140 api::utils::ivec4 input_tensor_dims = { in cumsum() local159 ivec4 input_tensor_dims; in cumsum() member
21 api::utils::ivec4& input_tensor_dims) { in set_softmax_kernel_params()135 api::utils::ivec4 input_tensor_dims = { in softmax_internal() local154 ivec4 input_tensor_dims; in softmax_internal() member
318 DSizes<Index, 5> input_tensor_dims = rnd_dims; in test_block_io_zero_stride() local
3550 const int* input_tensor_dims = input_tensor.dims->data; in VisitMediaPipeDeconvolutionNode() local4655 const int* input_tensor_dims = input_tensor.dims->data; in VisitTransposeConvNode() local
1539 llvm::SmallVector<int64_t> input_tensor_dims, in InferXlaConvOutputShape()