Searched defs:input_reshaped (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_diag_op.cc | 73 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
101 auto input_reshaped = input.flat_inner_dims<T, 2>(); in Compute() local
|
| D | split_op.cc | 135 const InputReshapedType& input_reshaped, in operator ()()
238 auto input_reshaped = in Compute() local
251 auto input_reshaped = input.shaped<T, 3>( in Compute() local
362 auto input_reshaped = in Compute() local
|
| D | split_v_op.cc | 189 const InputReshapedType& input_reshaped, in operator ()()
300 auto input_reshaped = in Compute() local
313 auto input_reshaped = input.shaped<T, 3>( in Compute() local
411 auto input_reshaped = input.shaped<T, 2>( in Compute() local
|
| D | unpack_op.cc | 106 auto input_reshaped = in Compute() local
|
| D | cholesky_op.cc | 136 auto input_reshaped = input.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
|
| D | matrix_set_diag_op.cc | 81 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
|
| D | matrix_band_part_op.cc | 59 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
|
| D | svd_op_gpu.cu.cc | 126 auto input_reshaped = M_copy.template flat_inner_dims<Scalar, 3>(); in RunSVD() local
|
| D | batch_kernels.cc | 149 auto input_reshaped = in SplitCPU() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | depthtospace_op.cc | 48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile() local
|
| D | spacetodepth_op.cc | 48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile() local
|