Searched refs:output_reshaped (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | cholesky_op.cc | 136 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
140 output_reshaped); in ComputeAsync()
162 output_reshaped_ptrs_base[batch] = &output_reshaped(batch, 0, 0); in ComputeAsync()
176 auto output_reshaped = output->template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
180 input_reshaped, output_reshaped); in ComputeAsync()
188 &output_reshaped(batch, 0, 0), n, in ComputeAsync()
|
| 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 | matrix_diag_op.cc | 135 auto output_reshaped = output->flat<T>(); in Compute() local
139 output_reshaped, lower_diag_index, upper_diag_index, max_diag_len, in Compute()
254 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute() local
258 output_reshaped, lower_diag_index, upper_diag_index, max_diag_len, in Compute()
|
| D | matrix_band_part_op.cc | 103 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute() local
106 input_reshaped, output_reshaped); in Compute()
|
| D | matrix_set_diag_op.cc | 162 auto output_reshaped = output->flat_inner_dims<T, 3>(); in Compute() local
165 output_reshaped, lower_diag_index, upper_diag_index, max_diag_len, 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()
|
| D | einsum_op_impl.h | 579 Tensor output_reshaped; in ContractOperands() local
581 ReshapeToRank3(*output, bcast.output_batch_size(), &output_reshaped)); in ContractOperands()
584 bcast, &output_reshaped); in ContractOperands()
|
| /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()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 779 output_reshaped = K.reshape(output_flat_transpose, [
782 return output_reshaped
|