Lines Matching defs:input_dims
137 const int input_dims = input.dims();
148 TensorShape output_shape({num_true_t(), input_dims});
166 switch (input_dims) {
176 "WhereOp : Unhandled input dimensions: ", input_dims));
256 const int input_dims = input.dims();
259 ComputeAsyncType<int32>(input, input_dims, context, done);
261 ComputeAsyncType<int64>(input, input_dims, context, done);
266 void ComputeAsyncType(const Tensor& input, const int input_dims,
301 auto create_and_check_output = [context, &d, &input, input_dims,
319 0, TensorShape({num_true, input_dims}), &output),
330 switch (input_dims) {
341 input_dims),
