Searched refs:out_reshaped (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 509 out_reshaped = op(inp_reshaped)
511 out_inner_shape = out_reshaped.shape[-inner_rank:]
513 out_inner_shape = array_ops.shape(out_reshaped)[-inner_rank:]
516 out_reshaped, array_ops.concat((batch_shape, out_inner_shape), axis=-1))
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | banded_triangular_solve_op.cc | 258 Tensor out_reshaped; in Compute() local
260 out_reshaped.CopyFrom(*out, TensorShape({batch_size, d1, d3})), in Compute()
265 &out_reshaped); in Compute()
|
| D | matrix_triangular_solve_op_impl.h | 204 Tensor out_reshaped;
206 out_reshaped.CopyFrom(*out, TensorShape({batch_size, d0, d3})),
211 &out_reshaped);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 719 Tensor out_reshaped;
721 out_reshaped.CopyFrom(*out, TensorShape({batch_size, d0, d3})),
733 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DT_FLOAT, out_reshaped.shape(),
748 out_reshaped.flat<bfloat16>().data(), out->NumElements());
752 trans_y_, bcast, &out_reshaped);
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 313 out_reshaped = op(inp_reshaped)
315 out_inner_shape = out_reshaped.shape[-inner_rank:]
317 out_inner_shape = array_ops.shape(out_reshaped)[-inner_rank:]
320 out_reshaped, array_ops.concat((batch_shape, out_inner_shape), axis=-1))
|