| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_band_part_op.cc | 59 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
75 context, num_lower <= input_reshaped.dimension(1), in Compute()
78 input_reshaped.dimension(1), ") got: ", num_lower)); in Compute()
85 OP_REQUIRES(context, num_upper <= input_reshaped.dimension(2), in Compute()
88 input_reshaped.dimension(2), in Compute()
92 ((num_lower < 0 || num_lower == input_reshaped.dimension(1)) && in Compute()
93 (num_upper < 0 || num_upper == input_reshaped.dimension(2)))) { in Compute()
105 input_reshaped, output_reshaped); in Compute()
|
| D | split_op.cc | 135 const InputReshapedType& input_reshaped, in operator ()() argument
161 use_parallelism_between_outputs, &input_reshaped, in operator ()()
180 result_shaped = input_reshaped.slice(slice_indices, slice_sizes); in operator ()()
185 input_reshaped, slice_indices, slice_sizes); in operator ()()
238 auto input_reshaped = in Compute() local
246 SplitOpCPUImpl<T, decltype(input_reshaped), 2>{}( in Compute()
247 context, input_reshaped, input_shape, split_dim, prefix_dim_size, in Compute()
251 auto input_reshaped = input.shaped<T, 3>( in Compute() local
261 SplitOpCPUImpl<T, decltype(input_reshaped), 3>{}( in Compute()
262 context, input_reshaped, input_shape, split_dim, prefix_dim_size, in Compute()
[all …]
|
| 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 | matrix_diag_op.cc | 73 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
76 context->eigen_device<Device>(), input_reshaped, output_reshaped); in Compute()
101 auto input_reshaped = input.flat_inner_dims<T, 2>(); in Compute() local
112 input_reshaped, output_reshaped); in Compute()
|
| D | cholesky_op.cc | 136 auto input_reshaped = input.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
140 n /* num_lower_diags */, 0 /* num_upper_diags */, input_reshaped, in ComputeAsync()
144 const int64 batch_size = input_reshaped.dimension(0); in ComputeAsync()
|
| D | matrix_set_diag_op.cc | 81 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
88 context, context->eigen_device<Device>(), input_reshaped, diag_reshaped, in Compute()
|
| D | unpack_op.cc | 106 auto input_reshaped = in Compute() local
119 output_shaped, input_reshaped, indices, in Compute()
|
| D | svd_op_gpu.cu.cc | 126 auto input_reshaped = M_copy.template flat_inner_dims<Scalar, 3>(); in RunSVD() local
127 input_ptr = input_reshaped.data(); in RunSVD()
133 const int64 batch_size = input_reshaped.dimension(0); in RunSVD()
|
| D | batch_kernels.cc | 149 auto input_reshaped = in SplitCPU() local
164 output_shaped, input_reshaped, in SplitCPU()
|
| /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()
|