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