Home
last modified time | relevance | path

Searched refs:CreateTensor (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op_test.cc76 {{{DatasetOpsTestBase::CreateTensor<int64>({2, 2}, {0, 0, 1, 1})}, in TestCases()
77 {DatasetOpsTestBase::CreateTensor<int32>({2}, {888, 999})}, in TestCases()
78 {DatasetOpsTestBase::CreateTensor<int64>({2}, {2, 2})}}, in TestCases()
79 {{{DatasetOpsTestBase::CreateTensor<int64>({1, 1}, {0})}, in TestCases()
80 {DatasetOpsTestBase::CreateTensor<int32>({1}, {888})}, in TestCases()
81 {DatasetOpsTestBase::CreateTensor<int64>({1}, {2})}}, in TestCases()
82 {{DatasetOpsTestBase::CreateTensor<int64>({1, 1}, {1})}, in TestCases()
83 {DatasetOpsTestBase::CreateTensor<int32>({1}, {999})}, in TestCases()
84 {DatasetOpsTestBase::CreateTensor<int64>({1}, {2})}}}, in TestCases()
86 {{{DatasetOpsTestBase::CreateTensor<int64>({2, 3}, {0, 0, 0, 1, 1, 1})}, in TestCases()
[all …]
Dconcatenate_dataset_op_test.cc83 {{DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
85 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
87 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
89 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
92 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {1, 2}), in TestCase1()
93 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {5, 6}), in TestCase1()
94 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {3, 4}), in TestCase1()
95 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {7, 8}), in TestCase1()
96 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {11, 12}), in TestCase1()
97 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {15, 16}), in TestCase1()
[all …]
Dtake_dataset_op_test.cc74 {DatasetOpsTestBase::CreateTensor<int64>( in TakeLessTestCase()
78 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {0}), in TakeLessTestCase()
79 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {1}), in TakeLessTestCase()
80 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {2}), in TakeLessTestCase()
81 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {3})}, in TakeLessTestCase()
92 {DatasetOpsTestBase::CreateTensor<int64>( in TakeMoreTestCase()
96 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {0}), in TakeMoreTestCase()
97 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {1}), in TakeMoreTestCase()
98 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {2}), in TakeMoreTestCase()
99 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {3}), in TakeMoreTestCase()
[all …]
Dtensor_slice_dataset_op_test.cc80 {{{DatasetOpsTestBase::CreateTensor<int64>(TensorShape({2}), {1, 2}), in TestCases()
81 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({2, 2}), in TestCases()
83 DatasetOpsTestBase::CreateTensor<double>(TensorShape({2, 1}), in TestCases()
85 DatasetOpsTestBase::CreateTensor<string>(TensorShape({2, 1}), in TestCases()
87 {{DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {1}), in TestCases()
88 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({2}), {1, 2}), in TestCases()
89 DatasetOpsTestBase::CreateTensor<double>(TensorShape({1}), {37.0}), in TestCases()
90 DatasetOpsTestBase::CreateTensor<string>(TensorShape({1}), {"a"}), in TestCases()
91 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {2}), in TestCases()
92 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({2}), {3, 4}), in TestCases()
[all …]
Drepeat_dataset_op_test.cc77 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, {1, 2, 3, 4}), in FiniteRepeatTestCase()
78 DatasetOpsTestBase::CreateTensor<string>(TensorShape{2, 1}, {"a", "b"})}, in FiniteRepeatTestCase()
81 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {1, 2}), in FiniteRepeatTestCase()
82 DatasetOpsTestBase::CreateTensor<string>(TensorShape{1}, {"a"}), in FiniteRepeatTestCase()
83 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {3, 4}), in FiniteRepeatTestCase()
84 DatasetOpsTestBase::CreateTensor<string>(TensorShape{1}, {"b"}), in FiniteRepeatTestCase()
85 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {1, 2}), in FiniteRepeatTestCase()
86 DatasetOpsTestBase::CreateTensor<string>(TensorShape{1}, {"a"}), in FiniteRepeatTestCase()
87 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {3, 4}), in FiniteRepeatTestCase()
88 DatasetOpsTestBase::CreateTensor<string>(TensorShape{1}, {"b"})}, in FiniteRepeatTestCase()
[all …]
Dtensor_dataset_op_test.cc82 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {1}), in PlainTensorsTestCase()
83 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({1, 3}), {1, 2, 3}), in PlainTensorsTestCase()
84 DatasetOpsTestBase::CreateTensor<double>(TensorShape({}), {37.0}), in PlainTensorsTestCase()
85 DatasetOpsTestBase::CreateTensor<string>(TensorShape({1, 2}), in PlainTensorsTestCase()
88 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {1}), in PlainTensorsTestCase()
89 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({1, 3}), {1, 2, 3}), in PlainTensorsTestCase()
90 DatasetOpsTestBase::CreateTensor<double>(TensorShape({}), {37.0}), in PlainTensorsTestCase()
91 DatasetOpsTestBase::CreateTensor<string>(TensorShape({1, 2}), in PlainTensorsTestCase()
106 {DatasetOpsTestBase::CreateTensor<Variant>( in NestedTensorsTestCase()
107 TensorShape({}), {DatasetOpsTestBase::CreateTensor<double>( in NestedTensorsTestCase()
[all …]
Dzip_dataset_op_test.cc93 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {0}), in TestCase1()
94 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {10}), in TestCase1()
95 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {1}), in TestCase1()
96 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {11}), in TestCase1()
97 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {2}), in TestCase1()
98 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {12})}, in TestCase1()
107 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {0}), in TestCase2()
108 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {10}), in TestCase2()
109 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {1}), in TestCase2()
110 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {11}), in TestCase2()
[all …]
Ddataset_test_base.h56 static Tensor CreateTensor(TensorShape input_shape, in CreateTensor() function
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_interpreter_test.cc110 CreateTensor(builder, in BuildMockModel()
114 CreateTensor(builder, in BuildMockModel()
118 CreateTensor(builder, in BuildMockModel()
Dsimple_tensor_allocator_test.cc67 const Offset<Tensor> tensor_offset = CreateTensor( in Create1dTensor()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dtypes_test.cc55 builder.Finish(CreateTensor(builder, 0, DataType::Serialize(T), in ToFlatBufferAndBack()
100 builder.Finish(::tflite::CreateTensor(builder)); in TEST()
125 builder.Finish(::tflite::CreateTensor(builder, 0, t, /*buffer*/ 1)); in TEST()
Dimport_test.cc58 ::tflite::CreateTensor(builder_, builder_.CreateVector<int>({1, 2, 2}), in BuildTensors()
62 ::tflite::CreateTensor(builder_, builder_.CreateVector<int>({2, 1}), in BuildTensors()
Dexport.cc42 using ::tflite::CreateTensor;
254 CreateTensor(*builder, builder->CreateVector(shape), type, buffer_index, in ExportTensors()
/external/tensorflow/tensorflow/lite/testing/
Dtf_driver.cc31 tensorflow::Tensor CreateTensor(const tensorflow::DataType type, in CreateTensor() function
189 auto tensor = CreateTensor(input_types_[id], input_shapes_[id]); in SetInput()
217 CreateTensor(input_types_[id], input_shapes_[id]); in ReshapeTensor()
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h373 CreateTensor(builder_, builder_.CreateVector<int>(t.shape), t.type, in AddTensorPerChannelQuant()
434 tensors_.push_back(CreateTensor(builder_,
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib.cc186 tensors.push_back(CreateTensor(*fbb, ExportVector<int32_t>(fbb, shape), in ExportTensors()
/external/tensorflow/tensorflow/lite/schema/
Dschema_generated.h2730 inline flatbuffers::Offset<Tensor> CreateTensor(
2756 return tflite::CreateTensor(
2766 flatbuffers::Offset<Tensor> CreateTensor(flatbuffers::FlatBufferBuilder &_fbb, const TensorT *_o, c…
8835 return CreateTensor(_fbb, _o, _rehasher);
8838 inline flatbuffers::Offset<Tensor> CreateTensor(flatbuffers::FlatBufferBuilder &_fbb, const TensorT…
8848 return tflite::CreateTensor(
11262 …ensor>> (_o->tensors.size(), [](size_t i, _VectorArgs *__va) { return CreateTensor(*__va->__fbb, _…