Searched refs:output_concat_dim (Results 1 – 4 of 4) sorted by relevance
122 int* output_concat_dim) { in CalculateConcatDims() argument129 *output_concat_dim = 0; in CalculateConcatDims()156 *output_concat_dim += in.dims() > 0 ? in.dim_size(concat_dim) : 1; in CalculateConcatDims()202 int output_concat_dim; in Compute() local204 inputs_flat_dim0, &inputs_flat, &output_concat_dim); in Compute()209 output_shape.AddDim(output_concat_dim); in Compute()211 output_shape.set_dim(concat_dim, output_concat_dim); in Compute()
114 int64 output_concat_dim = 0; in Compute() local142 output_concat_dim += in.dims() > 0 ? in.dim_size(axis) : 1; in Compute()148 output_shape.AddDim(output_concat_dim); in Compute()150 output_shape.set_dim(axis, output_concat_dim); in Compute()
101 int64 output_concat_dim = 0; in Compute() local119 output_concat_dim += in Compute()125 output_shape.AddDim(output_concat_dim); in Compute()127 output_shape.set_dim(axis, output_concat_dim); in Compute()
73 int output_concat_dim = 0; in Compile() local89 output_concat_dim += in_shape.dims() > 0 ? in_shape.dim_size(axis) : 1; in Compile()