| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_band_part_op.cc | 60 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
76 context, num_lower <= input_reshaped.dimension(1), in Compute()
79 input_reshaped.dimension(1), ") got: ", num_lower)); in Compute()
86 OP_REQUIRES(context, num_upper <= input_reshaped.dimension(2), in Compute()
89 input_reshaped.dimension(2), in Compute()
93 ((num_lower < 0 || num_lower == input_reshaped.dimension(1)) && in Compute()
94 (num_upper < 0 || num_upper == input_reshaped.dimension(2)))) { in Compute()
106 input_reshaped, output_reshaped); in Compute()
|
| D | cholesky_op.cc | 135 auto input_reshaped = input.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
139 n /* num_lower_diags */, 0 /* num_upper_diags */, input_reshaped, in ComputeAsync()
143 const int64 batch_size = input_reshaped.dimension(0); in ComputeAsync()
174 auto input_reshaped = const_cast<const Tensor*>(output) in ComputeAsync() local
180 input_reshaped, output_reshaped); in ComputeAsync()
|
| D | svd_op_gpu.cu.cc | 111 auto input_reshaped = M_copy.template flat_inner_dims<Scalar, 3>(); in RunSVD() local
112 input_ptr = input_reshaped.data(); in RunSVD()
114 M_copy.dims() > 2 ? input_reshaped.dimension(0) : 1; in RunSVD()
|
| D | matrix_set_diag_op.cc | 157 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
164 context, context->eigen_device<Device>(), input_reshaped, diag_reshaped, in Compute()
|
| D | matrix_diag_op.cc | 136 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
138 context, context->eigen_device<Device>(), input_reshaped, in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_op.cc | 132 const InputReshapedType& input_reshaped, in operator ()() argument
158 use_parallelism_between_outputs, &input_reshaped, in operator ()()
177 result_shaped = input_reshaped.slice(slice_indices, slice_sizes); in operator ()()
182 input_reshaped, slice_indices, slice_sizes); in operator ()()
235 auto input_reshaped = in Compute() local
243 SplitOpCPUImpl<T, decltype(input_reshaped), 2>{}( in Compute()
244 context, input_reshaped, input_shape, split_dim, prefix_dim_size, in Compute()
248 auto input_reshaped = input.shaped<T, 3>( in Compute() local
258 SplitOpCPUImpl<T, decltype(input_reshaped), 3>{}( in Compute()
259 context, input_reshaped, input_shape, split_dim, prefix_dim_size, in Compute()
|
| D | split_v_op.cc | 189 const InputReshapedType& input_reshaped, in operator ()() argument
214 use_parallelism_between_outputs, &input_reshaped, in operator ()()
233 result_shaped = input_reshaped.slice(current_indices, sizes); in operator ()()
238 input_reshaped, current_indices, sizes); in operator ()()
300 auto input_reshaped = in Compute() local
308 SplitVOpCPUImpl<T, Tlen, decltype(input_reshaped), 2>{}( in Compute()
309 context, input_reshaped, split_start_points, input_shape, split_dim, in Compute()
313 auto input_reshaped = input.shaped<T, 3>( in Compute() local
323 SplitVOpCPUImpl<T, Tlen, decltype(input_reshaped), 3>{}( in Compute()
324 context, input_reshaped, split_start_points, input_shape, split_dim, in Compute()
[all …]
|
| D | unpack_op.cc | 100 auto input_reshaped = in Compute() local
115 output_shaped, input_reshaped, indices, in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | depthtospace_op.cc | 48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile() local
49 OP_REQUIRES_OK(ctx, input_reshaped.status()); in Compile()
50 input = input_reshaped.ValueOrDie(); in Compile()
|
| D | spacetodepth_op.cc | 48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile() local
49 OP_REQUIRES_OK(ctx, input_reshaped.status()); in Compile()
50 input = input_reshaped.ValueOrDie(); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | concat_split_util.h | 164 auto input_reshaped = in SplitCPU() local
184 output_shaped, input_reshaped, in SplitCPU()
|