Home
last modified time | relevance | path

Searched refs:input_dtype (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dnot_supported.txt157 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/python/keras/layers/
Dembeddings_test.py45 input_dtype='int32',
54 input_dtype='int32',
63 input_dtype='int32',
73 input_dtype='int32',
/external/tensorflow/tensorflow/core/kernels/
Dguarantee_const_op.cc28 const DataType input_dtype = ctx->input_dtype(0); in Compute() local
29 OP_REQUIRES(ctx, input_dtype != DT_RESOURCE, in Compute()
Dqueue_op.cc55 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()
Dtraining_op_helpers.h113 if (ctx->input_dtype(input) == DT_RESOURCE) { in GetTrainingVariableMutex()
145 if (ctx->input_dtype(i) == DT_RESOURCE) { in MaybeLockVariableInputMutexesInOrder()
243 if (ctx->input_dtype(input) == DT_RESOURCE) { in GetInputTensorFromVariable()
Dlookup_table_init_op.cc54 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
122 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
Dstack.cc128 if (ctx->input_dtype(0) == DT_RESOURCE) { in GetStack()
226 if (ctx->input_dtype(1) != stack->ElemType()) { in ComputeAsync()
229 ctx->input_dtype(1))); in ComputeAsync()
Dtraining_op_helpers.cc25 if (ctx->input_dtype(input) != DT_RESOURCE) { in MaybeForwardRefInputToRefOutput()
Didentity_op.h28 if (IsRefType(context->input_dtype(0))) { in Compute()
Dcontrol_flow_ops.cc233 if (IsRefType(context->input_dtype(i))) { in Compute()
348 if (IsRefType(context->input_dtype(0))) { in Compute()
444 if (IsRefType(context->input_dtype(0))) { in Compute()
525 if (IsRefType(context->input_dtype(0))) { in Compute()
Dlookup_util.cc281 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype)); in GetLookupTable()
298 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype)); in GetInitializableLookupTable()
Dlookup_table_op.cc793 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
829 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
866 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
939 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF; in Compute()
Dlogging_ops.cc118 if (IsRefType(ctx->input_dtype(0))) { in Compute()
Dvariable_ops.cc143 CHECK(IsRefType(context->input_dtype(0))); in Compute()
/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py70 def layer_test(layer_cls, kwargs=None, input_shape=None, input_dtype=None, argument
95 if not input_dtype:
96 input_dtype = 'float32'
102 if input_dtype[:5] == 'float':
104 input_data = input_data.astype(input_dtype)
107 if input_dtype is None:
108 input_dtype = input_data.dtype
110 expected_output_dtype = input_dtype
126 x = keras.layers.Input(shape=input_shape[1:], dtype=input_dtype)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunique_dataset_op.cc38 DataType input_dtype = input->output_dtypes()[0]; in MakeDataset() local
40 input_dtype == DT_INT32 || input_dtype == DT_INT64 || in MakeDataset()
41 input_dtype == DT_STRING, in MakeDataset()
/external/tensorflow/tensorflow/lite/kernels/
Dcomparisons.cc62 template <typename input_dtype> \
92 GetTensorData<input_dtype>(input1), GetTensorShape(input2), \
93 GetTensorData<input_dtype>(input2), GetTensorShape(output), \
98 GetTensorData<input_dtype>(input1), GetTensorShape(input2), \
99 GetTensorData<input_dtype>(input2), GetTensorShape(output), \
/external/tensorflow/tensorflow/python/keras/utils/
Dlosses_utils.py214 input_dtype = losses.dtype
233 loss = math_ops.cast(loss, input_dtype)
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc25 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch() argument
77 xla::Pad(input, XlaHelpers::Zero(b, input_dtype), padding_config); in SpaceToBatch()
Dbatch_norm_op.cc127 DataType input_dtype = ctx->input_type(0); in Compile() local
210 ctx->SetOutput(0, XlaHelpers::ConvertElementType(x_backprop, input_dtype)); in Compile()
Dbatchtospace_op.cc25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace() argument
/external/tensorflow/tensorflow/contrib/losses/python/losses/
Dloss_ops.py103 input_dtype = losses.dtype
120 mean_loss = math_ops.cast(mean_loss, input_dtype)
/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
Dperiodic_resample_op.h393 const tensorflow::DataType input_tensor_type = context->input_dtype(0); in Compute()
414 const tensorflow::DataType grad_tensor_type = context->input_dtype(0); in Compute()
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Ddata_feeder_test.py57 for v in list(feeder.input_dtype.values()):
60 self.assertEqual(expected_np_dtype, feeder.input_dtype)
/external/tensorflow/tensorflow/python/ops/losses/
Dlosses_impl.py185 input_dtype = losses.dtype
206 loss = math_ops.cast(loss, input_dtype)

12