| /external/tensorflow/tensorflow/core/kernels/ |
| matrix_band_part_op.cc | 59 auto input_reshaped = input.flat_inner_dims<T, 3>(); variable
75 context, num_lower <= input_reshaped.dimension(1),
78 input_reshaped.dimension(1), ") got: ", num_lower));
85 OP_REQUIRES(context, num_upper <= input_reshaped.dimension(2),
88 input_reshaped.dimension(2),
92 ((num_lower < 0 || num_lower == input_reshaped.dimension(1)) &&
93 (num_upper < 0 || num_upper == input_reshaped.dimension(2)))) {
105 input_reshaped, output_reshaped);
|
| cholesky_op.cc | 136 auto input_reshaped = input.template flat_inner_dims<Scalar, 3>(); variable
140 n /* num_lower_diags */, 0 /* num_upper_diags */, input_reshaped,
144 const int64 batch_size = input_reshaped.dimension(0);
|
| split_op.cc | 135 const InputReshapedType& input_reshaped,
161 use_parallelism_between_outputs, &input_reshaped,
180 result_shaped = input_reshaped.slice(slice_indices, slice_sizes);
185 input_reshaped, slice_indices, slice_sizes);
238 auto input_reshaped =
246 SplitOpCPUImpl<T, decltype(input_reshaped), 2>{}(
247 context, input_reshaped, input_shape, split_dim, prefix_dim_size, member in class:SplitOpCPU
251 auto input_reshaped = input.shaped<T, 3>( member in class:__anon45533
261 SplitOpCPUImpl<T, decltype(input_reshaped), 3>{}(
262 context, input_reshaped, input_shape, split_dim, prefix_dim_size member in class:__anon45533
[all...] |
| split_v_op.cc | 189 const InputReshapedType& input_reshaped,
214 use_parallelism_between_outputs, &input_reshaped,
233 result_shaped = input_reshaped.slice(current_indices, sizes);
238 input_reshaped, current_indices, sizes);
300 auto input_reshaped =
308 SplitVOpCPUImpl<T, Tlen, decltype(input_reshaped), 2>{}(
309 context, input_reshaped, split_start_points, input_shape, split_dim,
313 auto input_reshaped = input.shaped<T, 3>(
323 SplitVOpCPUImpl<T, Tlen, decltype(input_reshaped), 3>{}(
324 context, input_reshaped, split_start_points, input_shape, split_dim member in class:tensorflow::SplitVOpCPU
[all...] |
| matrix_diag_op.cc | 73 auto input_reshaped = input.flat_inner_dims<T, 3>(); variable
76 context->eigen_device<Device>(), input_reshaped, output_reshaped);
101 auto input_reshaped = input.flat_inner_dims<T, 2>(); variable
112 input_reshaped, output_reshaped);
|
| matrix_set_diag_op.cc | 81 auto input_reshaped = input.flat_inner_dims<T, 3>(); variable
88 context, context->eigen_device<Device>(), input_reshaped, diag_reshaped,
|
| unpack_op.cc | 106 auto input_reshaped = variable
119 output_shaped, input_reshaped, indices,
|
| svd_op_gpu.cu.cc | 126 auto input_reshaped = M_copy.template flat_inner_dims<Scalar, 3>(); local
127 input_ptr = input_reshaped.data();
133 const int64 batch_size = input_reshaped.dimension(0);
|
| batch_kernels.cc | 149 auto input_reshaped = local
164 output_shaped, input_reshaped,
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| depthtospace_op.cc | 48 auto input_reshaped = NCHW_VECT_CToNCHW(input); variable
49 OP_REQUIRES_OK(ctx, input_reshaped.status());
50 input = input_reshaped.ValueOrDie();
|
| spacetodepth_op.cc | 48 auto input_reshaped = NCHW_VECT_CToNCHW(input); variable
49 OP_REQUIRES_OK(ctx, input_reshaped.status());
50 input = input_reshaped.ValueOrDie();
|
