| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge.py | 48 def _merge_function(self, inputs): argument
120 def call(self, inputs): argument
204 def compute_mask(self, inputs, mask=None): argument
250 def _merge_function(self, inputs): argument
289 def _merge_function(self, inputs): argument
319 def _merge_function(self, inputs): argument
356 def _merge_function(self, inputs): argument
386 def _merge_function(self, inputs): argument
416 def _merge_function(self, inputs): argument
521 def _merge_function(self, inputs): argument
[all …]
|
| /external/XNNPACK/eval/ |
| D | f32-exp.cc | 32 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
46 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
60 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
80 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
100 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
118 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
139 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
153 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
167 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
187 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
[all …]
|
| D | f32-expminus.cc | 32 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
46 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
60 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
80 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
98 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
119 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
133 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
147 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
167 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
185 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
[all …]
|
| D | f32-expm1minus.cc | 32 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
46 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
66 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
84 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
105 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
119 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
139 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
157 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
178 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
192 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
[all …]
|
| D | f32-roundne.cc | 28 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
46 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
64 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
82 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
100 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
112 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
124 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
142 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
160 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
178 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
[all …]
|
| D | f32-roundd.cc | 28 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
40 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
52 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
70 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
88 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
106 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
124 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
142 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
160 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
172 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
[all …]
|
| D | f32-roundu.cc | 28 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
40 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
52 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
70 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
88 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
106 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
124 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
142 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
160 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
172 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
[all …]
|
| D | f32-roundz.cc | 28 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
46 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
64 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
82 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
100 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
112 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
124 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
142 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
160 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
178 std::vector<float, AlignedAllocator<float, 64>> inputs(kBlockSize); in TEST() local
[all …]
|
| /external/libcxx/benchmarks/ |
| D | GenerateInput.hpp | 53 std::vector<IntT> inputs(N, static_cast<IntT>(-1)); in getDuplicateIntegerInputs() local
59 std::vector<IntT> inputs; in getSortedIntegerInputs() local
67 std::vector<IntT> inputs; in getSortedLargeIntegerInputs() local
76 std::vector<IntT> inputs = getSortedIntegerInputs<IntT>(N); in getSortedTopBitsIntegerInputs() local
83 std::vector<IntT> inputs; in getReverseSortedIntegerInputs() local
103 std::vector<IntT> inputs; in getRandomIntegerInputs() local
111 std::vector<std::string> inputs(N, getRandomString(1024)); in getDuplicateStringInputs() local
116 std::vector<std::string> inputs; in getRandomStringInputs() local
124 std::vector<std::string> inputs = getRandomStringInputs(N); in getSortedStringInputs() local
130 std::vector<std::string> inputs = getSortedStringInputs(N); in getReverseSortedStringInputs() local
[all …]
|
| /external/llvm-project/libcxx/benchmarks/ |
| D | GenerateInput.h | 59 std::vector<IntT> inputs; in getSortedIntegerInputs() local
67 std::vector<IntT> inputs; in getSortedLargeIntegerInputs() local
76 std::vector<IntT> inputs = getSortedIntegerInputs<IntT>(N); in getSortedTopBitsIntegerInputs() local
83 std::vector<IntT> inputs; in getReverseSortedIntegerInputs() local
103 std::vector<IntT> inputs; in getRandomIntegerInputs() local
116 std::vector<std::string> inputs; in getRandomStringInputs() local
124 std::vector<std::string> inputs = getRandomStringInputs(N); in getSortedStringInputs() local
130 std::vector<std::string> inputs = getSortedStringInputs(N); in getReverseSortedStringInputs() local
136 static std::vector<std::string> inputs = getRandomStringInputs(N); in getRandomCStringInputs() local
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | math_ops.cc | 29 Status Mul(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Mul()
41 absl::Span<AbstractTensorHandle* const> inputs, in Conj()
63 Status Add(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Add()
76 Status Sub(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Sub()
90 absl::Span<AbstractTensorHandle* const> inputs, in MatMul()
107 Status Neg(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Neg()
118 Status Sum(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Sum()
131 Status Div(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Div()
145 absl::Span<AbstractTensorHandle* const> inputs, in DivNoNan()
159 Status Exp(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Exp()
[all …]
|
| D | array_ops.cc | 26 absl::Span<AbstractTensorHandle* const> inputs, in Identity()
38 absl::Span<AbstractTensorHandle* const> inputs, in IdentityN()
50 absl::Span<AbstractTensorHandle* const> inputs, in ZerosLike()
61 absl::Span<AbstractTensorHandle* const> inputs, in Shape()
73 absl::Span<AbstractTensorHandle* const> inputs, in ExpandDims()
85 absl::Span<AbstractTensorHandle* const> inputs, in OnesLike()
|
| D | nn_ops.cc | 28 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in SparseSoftmaxCrossEntropyWithLogits()
45 absl::Span<AbstractTensorHandle* const> inputs, in ReluGrad()
60 absl::Span<AbstractTensorHandle* const> inputs, in Relu()
73 absl::Span<AbstractTensorHandle* const> inputs, in BiasAdd()
89 absl::Span<AbstractTensorHandle* const> inputs, in BiasAddGrad()
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | custom_training_loop_models_test.py | 77 def step_fn(inputs): argument
102 def step_fn(inputs): argument
140 def step_fn(inputs): argument
165 def step_fn(inputs): argument
197 def step_fn(inputs): argument
241 def step_fn(inputs): argument
293 def train_step_without_nested_tf_function(inputs): argument
295 def step_fn(inputs): argument
310 def train_step_with_nested_tf_function(inputs): argument
312 def step_fn(inputs): argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | base_test.py | 192 def call(self, inputs): argument
208 def call(self, inputs): argument
230 def call(self, inputs): argument
282 def call(self, inputs): argument
304 def call(self, inputs): argument
327 def call(self, inputs): argument
350 def call(self, inputs): argument
370 def call(self, inputs): argument
392 def call(self, inputs): argument
412 def call(self, inputs): argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | control_flow_test.py | 40 def call(self, inputs): argument
50 def call(self, inputs): argument
67 def call(self, inputs): argument
74 def call(self, inputs): argument
88 def call(self, inputs): argument
100 def call(self, inputs): argument
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | while_gradients_test.cc | 42 const std::vector<Output>* inputs = nullptr) { in CreateLoop()
95 [](const Scope& s, const std::vector<Output>& inputs, Output* output) { in TEST_F()
117 [](const Scope& s, const std::vector<Output>& inputs, Output* output) { in TEST_F()
122 std::vector<Output>* outputs) { in TEST_F()
168 [](const Scope& s, const std::vector<Output>& inputs, Output* output) { in TEST_F()
197 [](const Scope& s, const std::vector<Output>& inputs, Output* output) { in TEST_F()
202 std::vector<Output>* outputs) { in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | mnist_gradients_testutil.cc | 46 absl::Span<AbstractTensorHandle* const> inputs, in AddGradModel()
70 absl::Span<AbstractTensorHandle* const> inputs, in MatMulGradModel()
95 absl::Span<AbstractTensorHandle* const> inputs, in MNISTForwardModel()
150 absl::Span<AbstractTensorHandle* const> inputs, in MatMulTransposeModel()
163 absl::Span<AbstractTensorHandle* const> inputs, in MNISTGradModel()
216 absl::Span<AbstractTensorHandle* const> inputs, in ScalarMulModel()
224 absl::Span<AbstractTensorHandle* const> inputs, in MatMulModel()
233 absl::Span<AbstractTensorHandle* const> inputs, in MulModel()
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/layer_benchmarks/ |
| D | layer_benchmarks_test.py | 45 def _get_input_data(inputs): argument
217 def benchmark_layer_call(self, layer_cls, layer_args, inputs, num_iters): argument
228 self, layer_cls, layer_args, inputs, num_iters): argument
240 self, layer_cls, layer_args, inputs, num_iters): argument
256 self, layer_cls, layer_args, inputs, num_iters): argument
267 self, layer_cls, layer_args, inputs, num_iters): argument
279 self, layer_cls, layer_args, inputs, num_iters): argument
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | control_flow_test.cc | 29 Status LessThanTenCond(const Scope& scope, const std::vector<Output>& inputs, in LessThanTenCond()
35 Status AddOneBody(const Scope& scope, const std::vector<Output>& inputs, in AddOneBody()
41 Status NestedLoopBody(const Scope& scope, const std::vector<Output>& inputs, in NestedLoopBody()
50 std::vector<Output> inputs; in TEST() local
79 std::vector<Output> inputs; in TEST() local
122 std::vector<Output> inputs; in TEST() local
|
| /external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/ |
| D | NativeInterpreterWrapperTest.java | 126 Object[] inputs = {fourD}; in testRunWithFloat() local
144 Object[] inputs = {fourD}; in testRunWithBufferOutput() local
165 Object[] inputs = {fourD}; in testRunWithInputsOfSameDims() local
188 Object[] inputs = {fourD}; in testRunWithInt() local
206 Object[] inputs = {fourD}; in testRunWithLong() local
237 Object[] inputs = {fourD}; in testRunWithByte() local
270 Object[] inputs = {fourD}; in testRunWithString() local
290 Object[] inputs = {"s1"}; in testRunWithScalarString() local
304 Object[] inputs = {fourD}; in testRunWithString_supplementaryUnicodeCharacters() local
325 Object[] inputs = {fourD}; in testRunWithString_wrongShapeError() local
[all …]
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop_test.cc | 79 Status LessThanTenCond(const Scope& s, const std::vector<Output>& inputs, in LessThanTenCond()
85 Status AddOneBody(const Scope& s, const std::vector<Output>& inputs, in AddOneBody()
119 [](const Scope& s, const std::vector<Output>& inputs, Output* output) { in TEST_F()
134 [](const Scope& s, const std::vector<Output>& inputs, Output* output) { in TEST_F()
144 [](const Scope& s, const std::vector<Output>& inputs, Output* output) { in TEST_F()
168 [](const Scope& s, const std::vector<Output>& inputs, in TEST_F()
179 [](const Scope& s, const std::vector<Output>& inputs, in TEST_F()
194 [](const Scope& s, const std::vector<Output>& inputs, in TEST_F()
195 std::vector<Output>* outputs) { return s.status(); }, in TEST_F()
|
| /external/tflite-support/tensorflow_lite_support/codegen/ |
| D | code_generator_test.cc | 64 std::vector<std::string> inputs = {"image"}; in TEST_F() local
72 std::vector<std::string> inputs = {"image"}; in TEST_F() local
80 std::vector<std::string> inputs = {"image", "image"}; in TEST_F() local
88 std::vector<std::string> inputs = {"image", "image"}; in TEST_F() local
96 std::vector<std::string> inputs = {"image", "audio", "image", "audio", in TEST_F() local
110 std::vector<std::string> inputs = {"image", "image", "audio", "feature", in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_reachability.cc | 34 absl::Span<const HloInstruction* const> inputs, in SetReachabilityToUnion()
43 absl::Span<const HloInstruction* const> inputs, in FastSetReachabilityToUnion()
49 absl::Span<const HloInstruction* const> inputs, in SetReachabilityToUnionHelper()
84 std::vector<HloInstruction*> inputs; in BuildWithRestrictions() local
99 std::vector<HloInstruction*> inputs; in Build() local
162 std::vector<HloInstruction*> inputs; in UpdateReachabilityThroughInstruction() local
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | table_utils.py | 100 def _replace_oov_buckets(self, inputs, lookups): argument
117 def _lookup_and_mask(self, inputs): argument
140 def _ragged_lookup(self, inputs): argument
155 def _sparse_lookup(self, inputs): argument
166 def _tensor_lookup(self, inputs): argument
174 def lookup(self, inputs): argument
|