Home
last modified time | relevance | path

Searched refs:orig_input_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_op.cc46 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 Dbatchtospace_op.cc44 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 Dmkl_lrn_op.cc373 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 Dmkl_avgpooling_op.cc189 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 Dmkl_maxpooling_op.cc243 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 Dconv_ops_3d_test.py520 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)