Searched refs:orig_input_tensor (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | spacetobatch_op.cc | 46 const Tensor& orig_input_tensor, in SpaceToBatchOpCompute() argument 49 const int input_dims = orig_input_tensor.dims(); in SpaceToBatchOpCompute() 56 if (orig_input_tensor.dims() < 1 + block_dims) { in SpaceToBatchOpCompute() 58 " instead of ", orig_input_tensor.dims()); in SpaceToBatchOpCompute() 125 context->set_output(0, orig_input_tensor); in SpaceToBatchOpCompute() 141 orig_input_tensor.dim_size(0), block_shape_product); in SpaceToBatchOpCompute() 145 orig_input_tensor.dim_size(0), " and ", block_shape_product); in SpaceToBatchOpCompute() 149 int64_t input_batch_size = orig_input_tensor.dim_size(0); in SpaceToBatchOpCompute() 151 const int64_t size = orig_input_tensor.dim_size(block_dim + 1); in SpaceToBatchOpCompute() 165 const int64_t input_size = orig_input_tensor.dim_size(block_dim + 1); in SpaceToBatchOpCompute() [all …]
|
H A D | batchtospace_op.cc | 44 const Tensor& orig_input_tensor, in BatchToSpaceOpCompute() argument 47 const int input_dims = orig_input_tensor.dims(); in BatchToSpaceOpCompute() 55 context, orig_input_tensor.dims() >= 1 + block_dims, in BatchToSpaceOpCompute() 57 " instead of ", orig_input_tensor.dims())); in BatchToSpaceOpCompute() 106 const int64_t orig_input_batch_size = orig_input_tensor.dim_size(0); in BatchToSpaceOpCompute() 122 context->set_output(0, orig_input_tensor); in BatchToSpaceOpCompute() 141 const int64_t size = orig_input_tensor.dim_size(block_dim + 1); in BatchToSpaceOpCompute() 154 const int64_t input_size = orig_input_tensor.dim_size(block_dim + 1); in BatchToSpaceOpCompute() 169 const int64_t size = orig_input_tensor.dim_size(dim); in BatchToSpaceOpCompute() 194 orig_input_tensor.shaped<T, NUM_BLOCK_DIMS + 2>( \ in BatchToSpaceOpCompute() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
H A D | mkl_lrn_op.cc | 373 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in Compute() local 385 orig_input_dnn_data.SetUsrMem(orig_input_md, &orig_input_tensor); in Compute() 388 orig_input_dnn_data.SetUsrMemDataHandle(&orig_input_tensor, bwd_stream_); in Compute() 505 Tensor orig_input_tensor; in MklDefaultToEigen() local 526 orig_input_dnn_shape, &orig_input_tensor)); in MklDefaultToEigen() 528 orig_input_tensor = MklGetInput(context, kIdxOrigInput); in MklDefaultToEigen() 548 auto in_shaped = orig_input_tensor.shaped<T, 2>({nodes * batch, depth}); in MklDefaultToEigen() 593 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in SanityCheckInputs() local 617 OP_REQUIRES(context, orig_input_tensor.dims() == 4, in SanityCheckInputs()
|
H A D | mkl_avgpooling_op.cc | 189 const Tensor& orig_input_tensor = in Compute() local 204 auto shape_vec = orig_input_tensor.vec<int32>(); in Compute() 206 for (int i = 0; i < orig_input_tensor.NumElements(); i++) { in Compute()
|
H A D | mkl_maxpooling_op.cc | 243 const Tensor& orig_input_tensor = in Compute() local 260 TensorShape orig_input_shape = orig_input_tensor.shape(); in Compute()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
H A D | conv_ops_3d_test.py | 520 orig_input_tensor = constant_op.constant( 526 input_tensor = test_util.NHWCToNCHW(orig_input_tensor) 529 input_tensor = orig_input_tensor 547 orig_input_tensor, input_shape, conv, output_shape)
|