| /external/tensorflow/tensorflow/core/kernels/ |
| D | queue_op.cc | 99 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}), callback); in ComputeAsync()
140 OP_REQUIRES_OK_ASYNC(ctx, ctx->MatchSignature(expected_inputs, {}), callback); in ComputeAsync()
170 ctx, ctx->MatchSignature({DT_RESOURCE}, queue->component_dtypes()), in ComputeAsync()
174 ctx, ctx->MatchSignature({DT_STRING_REF}, queue->component_dtypes()), in ComputeAsync()
222 ctx->MatchSignature({DT_RESOURCE, DT_INT32}, queue->component_dtypes()), in ComputeAsync()
226 ctx->MatchSignature({DT_STRING_REF, DT_INT32}, in ComputeAsync()
295 ctx->MatchSignature({DT_RESOURCE, DT_INT32}, queue->component_dtypes()), in ComputeAsync()
299 ctx->MatchSignature({DT_STRING_REF, DT_INT32}, in ComputeAsync()
|
| D | sparse_conditional_accumulator_op.cc | 47 TF_RETURN_IF_ERROR(ctx->MatchSignature({}, {DT_STRING_REF})); in CheckSignature()
118 ctx->MatchSignature({DT_STRING_REF, DT_INT32}, in CheckSignature()
|
| D | conditional_accumulator_op.cc | 45 TF_RETURN_IF_ERROR(ctx->MatchSignature({}, {DT_STRING_REF})); in CheckSignature()
82 TF_RETURN_IF_ERROR(ctx->MatchSignature({}, {DT_RESOURCE})); in CheckSignature()
|
| D | lookup_table_init_op.cc | 59 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
127 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
|
| D | conditional_accumulator_base_op.h | 152 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in CheckSignature()
220 ctx->MatchSignature(GetExpectedInputs(accumulator), in CheckSignature()
|
| D | conditional_accumulator_base_op.cc | 98 ctx, ctx->MatchSignature(GetExpectedInputs(accumulator), {DT_INT32})); in CheckSignature()
|
| D | cwise_ops_common.cc | 24 OP_REQUIRES_OK(ctx, ctx->MatchSignature({in, in}, {out})); in BinaryOpShared()
|
| D | listdiff_op.cc | 33 OP_REQUIRES_OK(context, context->MatchSignature({dt, dt}, {dt, dtidx})); in ListDiffOp()
|
| D | clustering_ops.cc | 79 context->MatchSignature( in KmeansPlusPlusInitializationOp()
234 context->MatchSignature({DT_FLOAT, DT_INT64}, {DT_INT64})); in KMC2ChainInitializationOp()
286 context->MatchSignature({DT_FLOAT, DT_FLOAT, DT_INT64}, in NearestNeighborsOp()
|
| D | gather_nd_op.cc | 38 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t}, {dt})); in GatherNdOp()
|
| D | dense_update_ops.cc | 52 OP_REQUIRES_OK(context, context->MatchSignature({MakeRefType(dt), dt}, in DenseUpdateOp()
|
| D | scatter_nd_op.cc | 66 OP_REQUIRES_OK(c, c->MatchSignature({index_t, dt, index_t}, {dt})); in ScatterNdOp()
142 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t, dt}, {dt})); in TensorScatterOp()
244 OP_REQUIRES_OK(c, c->MatchSignature({dt_ref, index_t, dt}, {dt_ref})); in ScatterNdUpdateOp()
247 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t, dt}, {dt})); in ScatterNdUpdateOp()
|
| D | barrier_ops.cc | 542 ctx->MatchSignature({DT_STRING_REF, DT_STRING, in ComputeAsync()
598 ctx, ctx->MatchSignature(expected_inputs, expected_outputs), callback); in ComputeAsync()
|
| D | lookup_table_op.cc | 837 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
870 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
904 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
976 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
|
| D | reduction_ops_common.h | 136 OP_REQUIRES_OK(ctx, ctx->MatchSignature({dt, pt}, {dt}));
|
| D | dynamic_stitch_op.cc | 51 OP_REQUIRES_OK(c, c->MatchSignature(expected, {dt})); in DynamicStitchOpImplBase()
|
| D | control_flow_ops.cc | 204 OP_REQUIRES_OK(context, context->MatchSignature(DataTypeVector(num_in, dt), in MergeOp()
|
| D | cwise_ops_common.h | 298 OP_REQUIRES_OK(ctx, ctx->MatchSignature({in}, {out})); in UnaryOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | numeric_op.h | 34 OP_REQUIRES_OK(context, context->MatchSignature({dt}, {dt})); in UnaryOp()
44 OP_REQUIRES_OK(context, context->MatchSignature({dt, dt}, {dt})); in BinaryOp()
|
| D | op_kernel.h | 327 Status MatchSignature(const DataTypeSlice expected_inputs,
1168 Status MatchSignature(const DataTypeSlice expected_inputs,
|
| D | op_kernel.cc | 254 Status OpKernelConstruction::MatchSignature( in MatchSignature() function in tensorflow::OpKernelConstruction
1031 Status OpKernelContext::MatchSignature(const DataTypeSlice expected_inputs, in MatchSignature() function in tensorflow::OpKernelContext
|
| D | op_kernel_test.cc | 63 ::tensorflow::Status status = context->MatchSignature( in TestOp2()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bcast_ops.cc | 35 OP_REQUIRES_OK(ctx, ctx->MatchSignature({DT_INT32, DT_INT32}, {DT_INT32})); in BCastArgsOp()
83 ctx, ctx->MatchSignature({DT_INT32, DT_INT32}, {DT_INT32, DT_INT32})); in BCastGradArgsOp()
|
| D | dynamic_stitch_op.cc | 52 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected, {dt})); in DynamicStitchOp()
|