| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| depthtospace_op.cc | 172 auto output_reshaped = NCHWToNCHW_VECT_C(output); variable
173 OP_REQUIRES_OK(ctx, output_reshaped.status());
174 output = output_reshaped.ValueOrDie();
|
| spacetodepth_op.cc | 176 auto output_reshaped = NCHWToNCHW_VECT_C(output); variable
177 OP_REQUIRES_OK(ctx, output_reshaped.status());
178 output = output_reshaped.ValueOrDie();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| cholesky_op.cc | 137 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); variable
141 output_reshaped);
152 &output_reshaped(batch, 0, 0), n,
|
| matrix_diag_op.cc | 72 auto output_reshaped = output->flat_inner_dims<T, 2>(); variable
76 context->eigen_device<Device>(), input_reshaped, output_reshaped);
109 auto output_reshaped = output->flat_inner_dims<T, 3>(); variable
112 input_reshaped, output_reshaped);
|
| matrix_inverse_op.cc | 164 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); variable
178 output_ptr_array_base[batch] = &output_reshaped(batch, 0, 0);
228 eye(device, output_reshaped);
237 &output_reshaped(batch, 0, 0), n,
|
| matrix_set_diag_op.cc | 86 auto output_reshaped = output->flat_inner_dims<T, 3>(); variable
89 output_reshaped);
|
| matrix_band_part_op.cc | 102 auto output_reshaped = output->flat_inner_dims<T, 3>(); variable
105 input_reshaped, output_reshaped);
|
| determinant_op.cc | 200 auto output_reshaped = out->template flat_inner_dims<Scalar, 1>(); variable
240 pivots_mat.data(), output_reshaped, dev_info.back().mutable_data());
|
