| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | not_supported.txt | 157 mean/mean_axis=0,const_axis=True,input_dtype=tf.float32,input_shape=[3,3,2,4],keepdims=True
158 mean/mean_axis=0,const_axis=True,input_dtype=tf.float32,input_shape=[3,3,2,4],keepdims=False
159 mean/mean_axis=0,const_axis=False,input_dtype=tf.float32,input_shape=[3,3,2,4],keepdims=True
160 mean/mean_axis=0,const_axis=False,input_dtype=tf.float32,input_shape=[3,3,2,4],keepdims=False
161 mean/mean_axis=0,const_axis=True,input_dtype=tf.int32,input_shape=[3,3,2,4],keepdims=True
162 mean/mean_axis=0,const_axis=True,input_dtype=tf.int32,input_shape=[3,3,2,4],keepdims=False
163 mean/mean_axis=0,const_axis=False,input_dtype=tf.int32,input_shape=[3,3,2,4],keepdims=True
164 mean/mean_axis=0,const_axis=False,input_dtype=tf.int32,input_shape=[3,3,2,4],keepdims=False
165 mean/mean_axis=0,const_axis=True,input_dtype=tf.int64,input_shape=[3,3,2,4],keepdims=True
166 mean/mean_axis=0,const_axis=True,input_dtype=tf.int64,input_shape=[3,3,2,4],keepdims=False
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | guarantee_const_op.cc | 28 const DataType input_dtype = ctx->input_dtype(0); in Compute() local
29 OP_REQUIRES(ctx, input_dtype != DT_RESOURCE, in Compute()
|
| D | queue_op.cc | 55 if (ctx->input_dtype(0) == DT_RESOURCE) { in ComputeAsync()
91 if (ctx->input_dtype(0) == DT_RESOURCE) { in ComputeAsync()
132 if (ctx->input_dtype(0) == DT_RESOURCE) { in ComputeAsync()
168 if (ctx->input_dtype(0) == DT_RESOURCE) { in ComputeAsync()
219 if (ctx->input_dtype(0) == DT_RESOURCE) { in ComputeAsync()
292 if (ctx->input_dtype(0) == DT_RESOURCE) { in ComputeAsync()
|
| D | training_op_helpers.cc | 25 if (ctx->input_dtype(input) != DT_RESOURCE) { in MaybeForwardRefInputToRefOutput()
|
| D | training_op_helpers.h | 114 if (ctx->input_dtype(input) == DT_RESOURCE) { in GetTrainingVariableMutex()
146 if (ctx->input_dtype(i) == DT_RESOURCE) { in MaybeLockVariableInputMutexesInOrder()
242 if (ctx->input_dtype(input) == DT_RESOURCE) { in GetInputTensorFromVariable()
|
| D | control_flow_ops.cc | 219 if (IsRefType(context->input_dtype(i))) { in Compute()
296 if (IsRefType(context->input_dtype(0))) { in Compute()
353 if (IsRefType(context->input_dtype(0))) { in Compute()
405 if (IsRefType(context->input_dtype(0))) { in Compute()
|
| D | identity_op.h | 28 if (IsRefType(context->input_dtype(0))) { in Compute()
|
| D | lookup_table_init_op.cc | 55 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
124 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
|
| D | stack.cc | 129 if (ctx->input_dtype(0) == DT_RESOURCE) { in GetStack()
227 if (ctx->input_dtype(1) != stack->ElemType()) { in ComputeAsync()
230 ctx->input_dtype(1))); in ComputeAsync()
|
| D | lookup_util.cc | 303 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype)); in GetLookupTable()
315 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype)); in GetInitializableLookupTable()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 95 input_dtype=None, argument
139 if not input_dtype:
140 input_dtype = 'float32'
146 if input_dtype[:5] == 'float':
148 input_data = input_data.astype(input_dtype)
151 if input_dtype is None:
152 input_dtype = input_data.dtype
154 expected_output_dtype = input_dtype
192 x = layers.Input(shape=input_shape[1:], dtype=input_dtype)
229 tensor_spec.TensorSpec(shape=input_shape, dtype=input_dtype))
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings_test.py | 51 input_dtype='int32',
60 input_dtype='int32',
69 input_dtype='int32',
79 input_dtype='int32',
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | unique_dataset_op.cc | 219 DataType input_dtype = input->output_dtypes()[0]; in MakeDataset() local
221 input_dtype == DT_INT32 || input_dtype == DT_INT64 || in MakeDataset()
222 input_dtype == DT_STRING, in MakeDataset()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | comparisons.cc | 84 template <typename input_dtype, reference_ops::ComparisonFn<int32> opname>
110 reference_ops::BroadcastComparison4DSlowWithScaling<input_dtype, opname>( in ComparisonQuantized()
111 op_params, GetTensorShape(input1), GetTensorData<input_dtype>(input1), in ComparisonQuantized()
112 GetTensorShape(input2), GetTensorData<input_dtype>(input2), in ComparisonQuantized()
115 reference_ops::ComparisonWithScaling<input_dtype, opname>( in ComparisonQuantized()
116 op_params, GetTensorShape(input1), GetTensorData<input_dtype>(input1), in ComparisonQuantized()
117 GetTensorShape(input2), GetTensorData<input_dtype>(input2), in ComparisonQuantized()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | spacetobatch_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch() argument
77 xla::Pad(input, XlaHelpers::Zero(b, input_dtype), padding_config); in SpaceToBatch()
|
| D | batch_norm_op.cc | 253 DataType input_dtype = ctx->input_type(0); in Compile() local
336 ctx->SetOutput(0, XlaHelpers::ConvertElementType(x_backprop, input_dtype)); in Compile()
|
| D | batchtospace_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 123 const auto& input_dtype = input_tensor.dtype(); in ConvertTensor() local
126 TF_RETURN_IF_ERROR(ConvertDataType(input_dtype, *builder, &elt_type)); in ConvertTensor()
136 switch (input_dtype) { in ConvertTensor()
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | string_lookup_test.py | 93 input_dtype): argument
122 input_dtype=input_dtype,
|
| D | integer_lookup_test.py | 99 input_dtype): argument
128 input_dtype=input_dtype,
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | py_builtins.py | 472 input_dtype = getattr(input_, 'dtype', None)
474 if input_dtype != spec.dtype:
475 input_dtype_str = 'no dtype' if input_dtype is None else str(input_dtype)
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | losses_utils.py | 305 input_dtype = losses.dtype
322 loss = math_ops.cast(loss, input_dtype)
|
| D | composite_tensor_support_test.py | 127 input_dtype=None, argument
142 model.add(keras.Input(shape=input_shape, dtype=input_dtype))
154 inputs = keras.Input(shape=input_shape, dtype=input_dtype)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 184 DataType input_dtype = ctx.input_dtype(i); in ShapeTraceString() local
185 if (input_dtype == DataType::DT_RESOURCE || in ShapeTraceString()
186 input_dtype == DataType::DT_VARIANT || IsRefType(input_dtype)) { in ShapeTraceString()
191 DataTypeString(input_dtype), ctx.input(i).shape().DebugString())); in ShapeTraceString()
404 Status OpKernelContext::input_dtype(StringPiece name, DataType* dtype) const { in input_dtype() function in tensorflow::OpKernelContext
532 if (input_dtype(input_index) != output_dtype) { in forward_input()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | sequential.py | 275 input_dtype=None): argument
296 dtype=input_dtype,
|