Home
last modified time | relevance | path

Searched refs:input_t (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dencode_proto_op.cc78 auto input_t = input.flat_inner_dims<int64>(); in TotalPackedSize() local
81 input_t(static_cast<int64>(message_index), i)); in TotalPackedSize()
91 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize() local
94 input_t(static_cast<int64>(message_index), i)); in TotalPackedSize()
104 auto input_t = input.flat_inner_dims<int64>(); in TotalPackedSize() local
107 input_t(static_cast<int64>(message_index), i)); in TotalPackedSize()
117 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize() local
120 input_t(static_cast<int64>(message_index), i)); in TotalPackedSize()
155 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize() local
158 input_t(static_cast<int64>(message_index), i)); in TotalPackedSize()
[all …]
Dconv_grad_input_ops_benchmark_test.cc45 Tensor input_t = data_format == FORMAT_NHWC in Conv2DBackpropInput() local
59 TF_CHECK_OK(ComputeConv2DDimension(params, input_t, filter_t, &conv2d_dims)); in Conv2DBackpropInput()
69 input_dims_t.flat<int32>()(0) = input_t.dim_size(0); in Conv2DBackpropInput()
70 input_dims_t.flat<int32>()(1) = input_t.dim_size(1); in Conv2DBackpropInput()
71 input_dims_t.flat<int32>()(2) = input_t.dim_size(2); in Conv2DBackpropInput()
72 input_dims_t.flat<int32>()(3) = input_t.dim_size(3); in Conv2DBackpropInput()
Dpopulation_count_op.cc46 const Tensor& input_t = c->input(0); in Compute() local
48 OP_REQUIRES_OK(c, c->allocate_output(0, input_t.shape(), &output_t)); in Compute()
50 auto input = input_t.flat<T>(); in Compute()
Dcompare_and_bitpack_op.cc43 const Tensor& input_t = c->input(0); in Compute() local
49 const TensorShape& input_shape = input_t.shape(); in Compute()
66 auto input = input_t.flat_inner_dims<T>(); in Compute()
Dshape_ops.h143 const Tensor& input_t = ctx->input(0); in Compute() local
144 OP_REQUIRES(ctx, input_t.dtype() != DT_VARIANT, in Compute()
153 const TensorShape& input_shape = input_t.shape(); in Compute()
179 if (!output_t.CopyFrom(input_t, output_shape)) { in Compute()
Dconv_grad_filter_ops_benchmark_test.cc46 Tensor input_t = data_format == FORMAT_NHWC in Conv2DBackpropFilter() local
60 TF_CHECK_OK(ComputeConv2DDimension(params, input_t, filter_t, &conv2d_dims)); in Conv2DBackpropFilter()
75 Node* input = test::graph::Constant(graph, input_t, "input"); in Conv2DBackpropFilter()
Dslice_op.cc173 auto input_t = input.tensor<T, 2>(); in Compute() local
188 &input_t(row + 1, col_begin)); in Compute()
190 memcpy(&output_t(i, 0), &input_t(row, col_begin), in Compute()
Dxsmm_conv2d_test.cc138 LIBXSMM_VLA_DECL(4, const float, input_t, in naive_conv_fp()
155 LIBXSMM_VLA_ACCESS(4, input_t, img, ifm, ij + kj, ii + ki, in naive_conv_fp()
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/
Dindex_lookup_forward_benchmark.py76 input_t = keras.Input(shape=(), dtype=dtypes.string)
84 out_t = layer(input_t)
85 model = keras.Model(input_t, out_t)
108 input_t = keras.Input(shape=(), dtype=dtypes.string)
116 out_t = layer(input_t)
117 model = keras.Model(input_t, out_t)
Ddiscretization_adapt_benchmark.py53 input_t = keras.Input(shape=(1,))
55 _ = layer(input_t)
80 input_t = keras.Input(shape=(1,), dtype=dtypes.float32)
82 _ = layer(input_t)
Dindex_lookup_adapt_benchmark.py71 input_t = keras.Input(shape=(), dtype=dtypes.string)
78 _ = layer(input_t)
95 input_t = keras.Input(shape=(), dtype=dtypes.string)
102 _ = layer(input_t)
Dnormalization_adapt_benchmark.py64 input_t = keras.Input(shape=(1,))
66 _ = layer(input_t)
92 input_t = keras.Input(shape=(1,), dtype=dtypes.float32)
94 _ = layer(input_t)
Dcategory_encoding_benchmark.py41 input_t = keras.Input(shape=(sequence_length,), dtype=dtypes.int32)
44 _ = layer(input_t)
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc108 Tensor input_t = GetRandomInputTensor(dims); in DefaultConv2D() local
111 Node* input = test::graph::Constant(graph, input_t, "input"); in DefaultConv2D()
130 Tensor input_t = GetRandomInputTensor(dims); in MklConv2D() local
133 Node* input = test::graph::Constant(graph, input_t, "input"); in MklConv2D()
221 Tensor input_t = GetRandomInputTensor(dims); in DefaultConv2DBwdFilter() local
226 Node* input = test::graph::Constant(graph, input_t, "input"); in DefaultConv2DBwdFilter()
250 Tensor input_t = GetRandomInputTensor(dims); in MklConv2DBwdFilter() local
255 Node* input = test::graph::Constant(graph, input_t, "input"); in MklConv2DBwdFilter()
Dmkl_relu_op_test.cc53 Tensor input_t(DT_FLOAT, shape); in Activation() local
54 input_t.flat<float>().setRandom(); in Activation()
55 Node* input = test::graph::Constant(graph, input_t, "input"); in Activation()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc158 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
168 item.feed = {{"input", input_t}, in TEST_F()
270 auto input_t = GenerateRandomTensor<DT_FLOAT>(input_shape); in TEST_F() local
281 item.feed = {{"input", input_t}, {"scale", scale_t}, in TEST_F()
349 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F() local
355 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
497 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F() local
503 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
584 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F() local
590 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
[all …]
Dmkl_remapper_test.cc255 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); \
262 {"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; \
354 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
367 item.feed = {{"input", input_t}, {"scale", scale_t}, in TEST_F()
371 item.feed = {{"input", input_t}, in TEST_F()
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py959 input_t = constant_op.constant(2.0)
960 result_t = func(input_t)
962 [input_t], [result_t])
975 input_t = constant_op.constant(2.0)
976 result_t = func(input_t)
978 [input_t], [result_t])
990 input_t = constant_op.constant(2.0)
991 result_t = func(input_t)
993 [input_t], [result_t])
1002 input_t = array_ops.identity(var) * 5.0
[all …]
Drnn.py842 input_t = tuple(ta.read(time) for ta in input_ta)
844 for input_, shape in zip(input_t, inputs_got_shape):
847 input_t = tuple(ta[time.numpy()] for ta in input_ta)
849 input_t = nest.pack_sequence_as(structure=inputs, flat_sequence=input_t)
851 call_cell = lambda: cell(input_t, state)
Dgradients_util.py408 for input_t, placeholder_t in _Captures(t.op.graph):
410 return _MaybeCaptured(input_t)
474 for input_t, placeholder in _Captures(func):
475 if input_t is t:
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Doptimize_dataset_test.py144 input_t = array_ops.placeholder(dtypes.int32, (None, None, None))
145 dataset = dataset_ops.Dataset.from_tensors(input_t)
154 sess.run(init_op, {input_t: np.ones([512, 1024, 1025], np.int32)})
160 input_t = array_ops.placeholder(dtypes.int32, (None, None, None, None))
161 dataset = dataset_ops.Dataset.from_tensor_slices(input_t)
170 sess.run(init_op, {input_t: np.ones([1, 512, 1024, 1025], np.int32)})
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_eager_v1.py123 isinstance(input_t, (np.ndarray, float, int))
124 for input_t in nest.flatten(inputs)):
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dslice_op_test.py193 def func(input_t): argument
194 slice_t = input_t[:]
209 def func(input_t): argument
210 slice_t = input_t[1:, 1:]
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_test.cc413 TF_AbstractTensor* input_t = in TEST_P() local
417 TF_ExecuteOperation(fn_op, 1, &input_t, add_outputs, status.get()); in TEST_P()
432 TF_DeleteAbstractTensor(input_t); in TEST_P()
510 TF_AbstractTensor* input_t = in TEST_P() local
516 TF_ExecuteOperation(fn_op, 1, &input_t, mm_outputs, status.get()); in TEST_P()
539 TF_DeleteAbstractTensor(input_t); in TEST_P()
/external/e2fsprogs/lib/support/
Ddict.c1235 typedef char input_t[256]; typedef
1297 input_t in; in construct()
1318 if (!fgets(in, sizeof(input_t), stdin)) in construct()
1362 input_t in; in main()
1395 if (!fgets(in, sizeof(input_t), stdin)) in main()

12