Home
last modified time | relevance | path

Searched defs:input_t (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dencode_proto_op.cc78 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
91 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize() local
104 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
117 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize() local
155 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize() local
168 auto input_t = input.flat_inner_dims<uint32>(); in TotalPackedSize() local
181 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize() local
212 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize() local
224 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
236 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
[all …]
Dpopulation_count_op.cc46 const Tensor& input_t = c->input(0); in Compute() local
Dconv_grad_input_ops_benchmark_test.cc45 Tensor input_t = data_format == FORMAT_NHWC in Conv2DBackpropInput() local
Dconv_grad_filter_ops_benchmark_test.cc46 Tensor input_t = data_format == FORMAT_NHWC in Conv2DBackpropFilter() local
Dshape_ops.h143 const Tensor& input_t = ctx->input(0); in Compute() local
Dslice_op.cc173 auto input_t = input.tensor<T, 2>(); in Compute() local
Dconv_grad_filter_ops.cc163 auto input_t = input.tensor<T, 4>(); in operator ()() local
Dlist_kernels.h1077 auto input_t = input.flat_outer_dims<T, 2>(); in Compute() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc161 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
269 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
398 auto input_t = GenerateRandomTensor<DT_FLOAT>(input_shape); in TEST_F() local
535 auto input_t = GenerateRandomTensor<DT_FLOAT>(input_shape); in TEST_F() local
642 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 32, 3}); in RunTest() local
749 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 32, 3}); in RunTest() local
903 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 1, 3}); in RunTest() local
1036 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 4, 32, 32, 3}); in TEST_F() local
1102 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 4, 32, 32, 3}); in TEST_F() local
1168 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F() local
[all …]
Dmkl_remapper_test.cc353 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
581 auto input_t = GenerateRandomTensor<DTYPE>({8, 32, 32, 3}); in RelpaceAddWithBiasAddDepthwiseConv2D() local
782 Tensor input_t = Tensor(DataTypeToEnum<T>::v(), input_shape); in VerifyFused() local
Darithmetic_optimizer_test.cc1562 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F() local
1609 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F() local
1654 auto input_t = in TEST_F() local
1705 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F() local
Dconstant_folding_test.cc3223 auto input_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 4})); in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc105 Tensor input_t = GetRandomInputTensor(dims); in DefaultConv2D() local
126 Tensor input_t = GetRandomInputTensor(dims); in MklConv2D() local
214 Tensor input_t = GetRandomInputTensor(dims); in DefaultConv2DBwdFilter() local
242 Tensor input_t = GetRandomInputTensor(dims); in MklConv2DBwdFilter() local
Dmkl_swish_op_test.cc60 Tensor input_t(dtype, shape); in SwishGraph() local
Dmkl_relu_op_test.cc53 Tensor input_t(DT_FLOAT, shape); in Activation() local
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dslice_op_test.py188 def func(input_t): argument
204 def func(input_t): argument
/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
Dsimple_op.h144 const auto& input_t = input_t_or.value(); in ShapeInference() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_test.cc413 TF_AbstractTensor* input_t = in TEST_P() local
510 TF_AbstractTensor* input_t = in TEST_P() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_matrix.h635 const Tensor& input_t = ctx->input(index); in ExtractVariantFromInput() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc202 Tensor input_t = in TEST_F() local
228 Tensor input_t = ::tensorflow::test::AsTensor<int32>({1, 2, 3, 4, 5}); in TEST_F() local
254 Tensor input_t = ::tensorflow::test::AsTensor<int32>({1, 2, 3, 4, 5}); in TEST_F() local
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4242 def swap_batch_timestep(input_t): argument
4276 def _expand_mask(mask_t, input_t, fixed_dim=1): argument
4299 def _process_single_input_t(input_t): argument
/external/e2fsprogs/lib/support/
Ddict.c1246 typedef char input_t[256]; typedef
/external/f2fs-tools/fsck/
Ddict.c1204 typedef char input_t[256]; typedef
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py1612 def TestFn(input_t): argument