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()200 int output_concat_dim; in Compute() local202 inputs_flat_dim0, &inputs_flat, &output_concat_dim); in Compute()207 output_shape.AddDim(output_concat_dim); in Compute()209 output_shape.set_dim(concat_dim, output_concat_dim); in Compute()
121 int64 output_concat_dim = 0; in Compute() local147 output_concat_dim += in.dims() > 0 ? in.dim_size(axis) : 1; in Compute()153 output_shape.AddDim(output_concat_dim); in Compute()155 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()
180 int64 output_concat_dim = 0; in Compute() local198 output_concat_dim += in Compute()204 output_shape.AddDim(output_concat_dim); in Compute()206 output_shape.set_dim(axis, output_concat_dim); in Compute()