| /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 | conditional_accumulator_base_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->MatchSignature({DT_STRING_REF, DT_INT64}, {})); in Compute()
74 OP_REQUIRES_OK(ctx, ctx->MatchSignature({DT_STRING_REF}, {DT_INT32})); in Compute()
|
| D | conditional_accumulator_op.cc | 78 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in CheckSignature()
105 ctx->MatchSignature({DT_STRING_REF, DT_INT32}, {accumulator->dtype()}), in CheckSignature()
|
| D | sparse_conditional_accumulator_op.cc | 79 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in CheckSignature()
107 ctx->MatchSignature({DT_STRING_REF, DT_INT32}, in CheckSignature()
|
| D | lookup_table_init_op.cc | 58 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
125 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
|
| D | cwise_ops_common.cc | 24 OP_REQUIRES_OK(ctx, ctx->MatchSignature({in, in}, {out})); in BinaryOpShared()
|
| 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 | scatter_nd_op.cc | 71 OP_REQUIRES_OK(c, c->MatchSignature({index_t, dt, index_t}, {dt})); in ScatterNdOp()
145 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t, dt}, {dt})); in TensorScatterOp()
236 OP_REQUIRES_OK(c, c->MatchSignature({dt_ref, index_t, dt}, {dt_ref})); in ScatterNdUpdateOp()
239 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t, dt}, {dt})); in ScatterNdUpdateOp()
|
| D | listdiff_op.cc | 33 OP_REQUIRES_OK(context, context->MatchSignature({dt, dt}, {dt, dtidx})); in ListDiffOp()
|
| D | dense_update_ops.cc | 52 OP_REQUIRES_OK(context, context->MatchSignature({MakeRefType(dt), dt}, in DenseUpdateOp()
|
| D | conditional_accumulator_base_op.h | 95 TF_RETURN_IF_ERROR(ctx->MatchSignature(expected_inputs, {DT_STRING_REF})); in SetAccumulatorHandle()
|
| D | lookup_table_op.cc | 797 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, expected_outputs)); in Compute()
832 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
868 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
942 OP_REQUIRES_OK(ctx, ctx->MatchSignature(expected_inputs, {})); in Compute()
|
| 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 | gather_nd_op.cc | 41 OP_REQUIRES_OK(c, c->MatchSignature({dt, index_t}, {dt})); in GatherNdOp()
|
| D | reduction_ops_common.h | 143 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 | 218 OP_REQUIRES_OK(context, context->MatchSignature(DataTypeVector(num_in, dt), in MergeOp()
|
| D | cwise_ops_common.h | 237 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 | 331 Status MatchSignature(const DataTypeSlice expected_inputs,
1131 Status MatchSignature(const DataTypeSlice expected_inputs,
|
| D | op_kernel.cc | 229 Status OpKernelConstruction::MatchSignature( in MatchSignature() function in tensorflow::OpKernelConstruction
872 Status OpKernelContext::MatchSignature(const DataTypeSlice expected_inputs, in MatchSignature() function in tensorflow::OpKernelContext
|
| /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()
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | masked_matmul_ops.cc | 63 context->MatchSignature( in MaskedMatmulOp()
|
| D | wals_solver_ops.cc | 66 context->MatchSignature( in WALSComputePartialLhsAndRhsOp()
|