| /external/tensorflow/tensorflow/core/kernels/ |
| guarantee_const_op.cc | 28 const DataType input_dtype = ctx->input_dtype(0); variable
29 OP_REQUIRES(ctx, input_dtype != DT_RESOURCE,
|
| identity_op.h | 28 if (IsRefType(context->input_dtype(0))) {
|
| training_op_helpers.cc | 21 if (ctx->input_dtype(input) == DT_RESOURCE) {
70 if (ctx->input_dtype(input) != DT_RESOURCE) {
|
| queue_ops.cc | 36 if (ctx->input_dtype(0) == DT_RESOURCE) {
86 if (ctx->input_dtype(0) == DT_RESOURCE) {
138 if (ctx->input_dtype(0) == DT_RESOURCE) {
187 if (ctx->input_dtype(0) == DT_RESOURCE) {
248 if (ctx->input_dtype(0) == DT_RESOURCE) {
334 if (ctx->input_dtype(0) == DT_RESOURCE) {
|
| lookup_table_init_op.cc | 54 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF;
124 (ctx->input_dtype(0) == DT_RESOURCE) ? DT_RESOURCE : DT_STRING_REF;
|
| stateless_random_ops.cc | 55 if (context->input_dtype(1) == DT_INT32) {
60 CHECK_EQ(DT_INT64, context->input_dtype(1));
|
| logging_ops.cc | 64 if (IsRefType(ctx->input_dtype(0))) {
|
| training_op_helpers.h | 65 if (ctx->input_dtype(input) == DT_RESOURCE) {
|
| control_flow_ops.cc | 226 if (IsRefType(context->input_dtype(i))) {
337 if (IsRefType(context->input_dtype(0))) {
432 if (IsRefType(context->input_dtype(0))) {
511 if (IsRefType(context->input_dtype(0))) {
|
| lookup_util.cc | 269 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype));
286 TF_RETURN_IF_ERROR(ctx->input_dtype(input_name, &handle_dtype));
|
| stack_ops.cc | 135 if (ctx->input_dtype(0) == DT_RESOURCE) {
260 if (ctx->input_dtype(1) != stack->ElemType()) {
263 ctx->input_dtype(1)));
|
| ops_testutil.h | 191 CHECK(!IsRefType(context_->input_dtype(input_index)));
|
| variable_ops.cc | 146 CHECK(IsRefType(context->input_dtype(0)));
|
| scatter_nd_op.cc | 113 DCHECK(IsRefType(c->input_dtype(0)));
145 } else if (IsRefType(c->input_dtype(0))) {
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/ |
| embeddings_test.py | 36 input_dtype='int32',
46 input_dtype='int32',
56 input_dtype='int32',
67 input_dtype='int32',
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| parallel_check_op.cc | 81 CHECK_EQ(ctx->input_dtype(i), ctx->input_dtype(i + num_pairs));
91 switch (ctx->input_dtype(i)) {
94 CompareTensors<int32>(ctx->input_dtype(i), v0, v1, num_elts, i);
98 CompareTensors<int64>(ctx->input_dtype(i), v0, v1, num_elts, i);
102 CompareTensors<float>(ctx->input_dtype(i), v0, v1, num_elts, i);
106 CompareTensors<double>(ctx->input_dtype(i), v0, v1, num_elts, i);
110 CompareTensors<bool>(ctx->input_dtype(i), v0, v1, num_elts, i);
113 LOG(FATAL) << "unimpl: " << ctx->input_dtype(i);
129 if (IsRefType(ctx->input_dtype(i)))
[all...] |
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| testing_utils.py | 54 def layer_test(layer_cls, kwargs=None, input_shape=None, input_dtype=None,
64 input_dtype: Data type of the input data.
75 if not input_dtype:
76 input_dtype = 'float32'
82 if input_dtype[:5] == 'float':
84 input_data = input_data.astype(input_dtype)
87 if input_dtype is None:
88 input_dtype = input_data.dtype
90 expected_output_dtype = input_dtype
106 x = keras.layers.Input(shape=input_shape[1:], dtype=input_dtype)
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/ |
| unique_dataset_op.cc | 38 DataType input_dtype = input->output_dtypes()[0]; variable
40 input_dtype == DT_INT32 || input_dtype == DT_INT64 ||
41 input_dtype == DT_STRING,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batch_norm_op.cc | 133 DataType input_dtype = ctx->input_type(0); variable
136 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_dtype, &input_type));
|
| spacetobatch_op.cc | 24 const xla::ComputationDataHandle& input, DataType input_dtype,
77 b->Pad(input, XlaHelpers::Zero(b, input_dtype), padding_config);
|
| batchtospace_op.cc | 24 const xla::ComputationDataHandle& input, DataType input_dtype,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_compilation_cache.cc | 112 signature->arg_types.emplace_back(ctx->input_dtype(input_num),
249 VLOG(2) << i << ": dtype=" << DataTypeString(ctx->input_dtype(i))
|
| /external/tensorflow/tensorflow/contrib/losses/python/losses/ |
| loss_ops.py | 128 input_dtype = losses.dtype
145 mean_loss = math_ops.cast(mean_loss, input_dtype)
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| periodic_resample_op.h | 231 const tensorflow::DataType input_tensor_type = context->input_dtype(0);
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| data_feeder_test.py | 48 for k, v in list(feeder.input_dtype.items()):
51 self.assertEqual(expected_np_dtype, feeder.input_dtype)
|
