Home
last modified time | relevance | path

Searched refs:data_tensor (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dstring_ngrams_op_test.py34 data_tensor = ragged_factory_ops.constant(data)
36 data_tensor, ngram_width=3, separator=b"|")
43 data_tensor = ragged_factory_ops.constant(data)
45 data_tensor, ngram_width=(2, 3), separator=b"|")
53 data_tensor = ragged_factory_ops.constant(data)
55 data_tensor, ngram_width=(3, 2), separator=b"|")
63 data_tensor = ragged_factory_ops.constant(data)
65 data_tensor, ngram_width=[2, 3], separator=b"|")
73 data_tensor = ragged_factory_ops.constant(data)
75 data_tensor, ngram_width=[3, 2], separator=b"|")
[all …]
H A Dragged_dynamic_partition_op_test.py165 data_tensor = ragged_factory_ops.constant(
177 data_tensor, segment_ids_tensor, num_partitions)
185 data_flow_ops.dynamic_partition(data_tensor, segment_ids_tensor,
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_reduction_sycl.cpp68 using data_tensor = Tensor<DataType, 2, DataLayout, IndexType>; in test_full_reductions_sum_with_offset_sycl() typedef
75 data_tensor in(tensor_range); in test_full_reductions_sum_with_offset_sycl()
84 TensorMap<data_tensor> in_offset(in.data() + offset, tensor_offset_range); in test_full_reductions_sum_with_offset_sycl()
92 TensorMap<data_tensor> in_gpu(gpu_in_data + offset, tensor_offset_range); in test_full_reductions_sum_with_offset_sycl()
142 using data_tensor = Tensor<DataType, 2, DataLayout, IndexType>; in test_full_reductions_max_with_offset_sycl() typedef
149 data_tensor in(tensor_range); in test_full_reductions_max_with_offset_sycl()
161 TensorMap<data_tensor> in_offset(in.data() + offset, tensor_offset_range); in test_full_reductions_max_with_offset_sycl()
170 TensorMap<data_tensor> in_gpu(gpu_in_data + offset, tensor_offset_range); in test_full_reductions_max_with_offset_sycl()
275 using data_tensor = Tensor<DataType, 2, DataLayout, IndexType>; in test_full_reductions_mean_with_offset_sycl() typedef
282 data_tensor in(tensor_range); in test_full_reductions_mean_with_offset_sycl()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dunsorted_segment.py93 data_tensor = tf.compat.v1.placeholder(
102 output = unsorted_segment_op(data_tensor, segment_ids_tensor,
104 return [data_tensor], [output]
109 data_tensor = tf.compat.v1.placeholder(
120 intermediate_tensor = unsorted_segment_op(data_tensor, segment_ids_tensor,
131 return [data_tensor, segment_ids_tensor], [output]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_builder.cc126 const TfLiteIntArray* inputs, const TfLiteTensor& data_tensor, in BuildDilatedDwConv() argument
143 &input_depth_size, data_tensor.dims); in BuildDilatedDwConv()
240 const TfLiteIntArray* inputs, const TfLiteTensor& data_tensor, in BuildSplittedDwConv() argument
249 &input_depth_size, data_tensor.dims); in BuildSplittedDwConv()
375 const auto& data_tensor = context->tensors[inputs->data[0]]; in PopulateSubGraph() local
379 &input_depth_size, data_tensor.dims); in PopulateSubGraph()
383 ComputeMinAndMaxQuantValues(data_tensor, &data_min, &data_max)); in PopulateSubGraph()
479 BuildDilatedDwConv(inputs, data_tensor, output_data_tensor, data_min_const, in PopulateSubGraph()
485 BuildSplittedDwConv(inputs, data_tensor, output_data_tensor, data_min_const, in PopulateSubGraph()
H A Dconv_2d_builder.h76 const TfLiteTensor& data_tensor,
86 const TfLiteIntArray* inputs, const TfLiteTensor& data_tensor,
153 const TfLiteTensor& data_tensor, const TfLiteTensor& bias_tensor,
H A Dtranspose_conv_2d_builder.cc44 const auto& data_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
86 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, data_tensor)); in PopulateSubGraph()
152 ProcessPerChannelQuantizedBias(data_tensor, bias_tensor, inputs->data[3], in PopulateSubGraph()
H A Dpool_2d_builder.cc34 const auto& data_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
36 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, data_tensor)); in PopulateSubGraph()
H A Dconcat_builder.cc54 const auto& data_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
57 ComputeMinAndMaxQuantValues(data_tensor, &data_min, &data_max)); in PopulateSubGraph()
H A Dmatmul_builder.cc55 const auto& data_tensor = context->tensors[data_tensor_id]; in AddFullyConnectedHelper() local
58 data_tensor, &data_min, &data_max)); in AddFullyConnectedHelper()
H A Dconv_2d_helpers.cc102 const TfLiteTensor& data_tensor, const TfLiteTensor& bias_tensor, in ProcessPerChannelQuantizedBias() argument
109 data_tensor.quantization.params); in ProcessPerChannelQuantizedBias()
/aosp_15_r20/external/tensorflow/tensorflow/examples/speech_commands/
H A Dinput_data.py599 summary, data_tensor = sess.run(
602 data[i - offset, :] = data_tensor.flatten()
632 data_tensor = sess.run([self.output_], feed_dict=input_dict)
633 return data_tensor
H A Dtest_streaming_accuracy.py134 data_tensor = sess.graph.get_tensor_by_name(FLAGS.input_names[0])
145 data_tensor:
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A Dstage.py1134 data_tensor = torch.cat(data)
1135 dt_shape = data_tensor.shape[0]
1140 metadata_tensor[:dt_shape] = data_tensor