| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | shape_output_test.py | 57 input_mask=[[False, True, True, True]],
88 input_mask=[[False, True, True, True]],
117 input_mask=[[False, True, True, True]],
149 input_mask=[[False, True, True, True]],
183 input_mask=[[False, True, True, True], [False, True]],
221 input_mask=[[True], []],
|
| D | trt_mode_test.py | 103 input_mask=[[True, True, True]],
127 input_mask=[[False, False, False]],
|
| D | batch_matmul_test.py | 40 input_mask=[[True] * len(s) for s in input_shapes],
|
| D | tf_trt_integration_test_base.py | 200 input_mask = [[False] + [True] * (len(shape) - 1) for shape in input_shapes]
205 output_shapes, input_mask, output_mask, [],
209 input_mask, output_mask, extra_inputs, extra_outputs): argument
243 assert len(input_mask) == len(input_shapes), (
261 for i, (shape, mask) in enumerate(zip(input_shapes, input_mask))
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_shape_optimization_profiles.h | 87 const std::vector<bool>& input_mask) const { in SetDimensions()
91 if (input_mask.size() != n_inputs_tf) { in SetDimensions()
92 return errors::Internal("Incorrect input mask size: ", input_mask.size()); in SetDimensions()
95 for (bool mask_val : input_mask) { in SetDimensions()
107 if (input_mask[i]) { in SetDimensions()
216 void SetInputMask(const std::vector<bool>& input_mask) { in SetInputMask() argument
217 input_mask_ = input_mask; in SetInputMask()
|
| D | trt_shape_optimization_profiles_test.cc | 192 std::vector<bool> input_mask(2, true); in TEST_P() local
193 profile.SetInputMask(input_mask); in TEST_P()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| D | bert_question_answerer.cc | 267 std::vector<int> input_mask; in Preprocess() local
268 input_mask.reserve(kMaxSeqLen); in Preprocess()
269 input_mask.insert(input_mask.end(), tokens.size(), 1); in Preprocess()
273 input_mask.insert(input_mask.end(), zeros_to_pad, 0); in Preprocess()
279 PopulateTensor(input_mask, mask_tensor); in Preprocess()
|
| /external/kernel-headers/original/uapi/linux/ |
| D | input.h | 126 struct input_mask { struct
218 #define EVIOCGMASK _IOR('E', 0x92, struct input_mask) /* Get event-masks */
241 #define EVIOCSMASK _IOW('E', 0x93, struct input_mask) /* Set event-masks */
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | bert_nl_classifier.cc | 110 std::vector<int> input_mask(input_tensor_length, 0); in Preprocess() local
114 input_mask[i] = 1; in Preprocess()
122 PopulateTensor(input_mask, mask_tensor); in Preprocess()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 103 std::vector<bool> input_mask = {true}; local
104 profile.SetInputMask(input_mask);
|
| D | trt_engine_resource_ops_test.cc | 166 std::vector<bool> input_mask(n_input, true); in CreateTRTEngine() local
167 profile.SetInputMask(input_mask); in CreateTRTEngine()
|
| /external/vulkan-validation-layers/tests/ |
| D | vktestbinding.h | 426 …VkBufferMemoryBarrier buffer_memory_barrier(VkFlags output_mask, VkFlags input_mask, VkDeviceSize … in buffer_memory_barrier() argument
432 barrier.dstAccessMask = input_mask; in buffer_memory_barrier()
505 …VkImageMemoryBarrier image_memory_barrier(VkFlags output_mask, VkFlags input_mask, VkImageLayout o…
512 barrier.dstAccessMask = input_mask;
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.metrics.-one-hot-io-u.pbtxt | 42 name: "input_mask"
|
| D | tensorflow.metrics.-precision-at-recall.pbtxt | 41 name: "input_mask"
|
| D | tensorflow.metrics.-binary-io-u.pbtxt | 42 name: "input_mask"
|
| D | tensorflow.metrics.-metric.pbtxt | 39 name: "input_mask"
|
| D | tensorflow.metrics.-squared-hinge.pbtxt | 43 name: "input_mask"
|
| D | tensorflow.metrics.-binary-crossentropy.pbtxt | 43 name: "input_mask"
|
| D | tensorflow.metrics.-hinge.pbtxt | 43 name: "input_mask"
|
| D | tensorflow.metrics.-sum.pbtxt | 41 name: "input_mask"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.layers.-average-pooling1-d.pbtxt | 47 name: "input_mask"
|
| D | tensorflow.layers.-average-pooling3-d.pbtxt | 47 name: "input_mask"
|
| D | tensorflow.layers.-max-pooling1-d.pbtxt | 47 name: "input_mask"
|
| D | tensorflow.layers.-dropout.pbtxt | 47 name: "input_mask"
|
| D | tensorflow.layers.-flatten.pbtxt | 46 name: "input_mask"
|