Searched refs:input_dims_to_stack (Results 1 – 1 of 1) sorted by relevance
49 const int64 input_dims_to_stack = input.dims() - 1; in Compute() local51 for (int i = 0; i < input_dims_to_stack; ++i) { in Compute()55 if (num_sparse_tensors == 1 && input_dims_to_stack == 0) { in Compute()108 2, {input_dims_to_stack + element_shape->NumElements()}, in Compute()113 for (int64 j = 0; j < input_dims_to_stack; ++j) { in Compute()117 output_shape_vec(j + input_dims_to_stack) = element_shape_vec(j); in Compute()136 output_shape->NumElements() - input_dims_to_stack == in Compute()141 i, "] was: ", output_shape->NumElements() - input_dims_to_stack, in Compute()146 output_shape_vec(j + input_dims_to_stack) = std::max( in Compute()147 output_shape_vec(j + input_dims_to_stack), element_shape_vec(j)); in Compute()[all …]