Searched refs:argument_t (Results 1 – 8 of 8) sorted by relevance
40 const Tensor* argument_t; in ParseScalarArgument() local41 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseScalarArgument()42 if (!TensorShapeUtils::IsScalar(argument_t->shape())) { in ParseScalarArgument()45 *output = argument_t->scalar<T>()(); in ParseScalarArgument()
1127 const Tensor* argument_t; in ParseScalarArgument() local1128 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseScalarArgument()1129 if (!TensorShapeUtils::IsScalar(argument_t->shape())) { in ParseScalarArgument()1132 *output = argument_t->scalar<T>()(); in ParseScalarArgument()1140 const Tensor* argument_t; in ParseVectorArgument() local1141 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseVectorArgument()1142 if (!TensorShapeUtils::IsVector(argument_t->shape())) { in ParseVectorArgument()1145 int size = argument_t->vec<T>().size(); in ParseVectorArgument()1148 output->push_back(argument_t->vec<T>()(i)); in ParseVectorArgument()
222 } argument_t; typedef
77 static const argument_t argument_mapping[] =195 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()218 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()
282 }argument_t; typedef284 static const argument_t argument_mapping[] =1077 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()1114 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()
281 }argument_t; typedef283 static const argument_t argument_mapping[] =1095 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()1132 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()
283 } argument_t; typedef285 static const argument_t argument_mapping[] =1093 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()1130 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()
134 } argument_t; typedef136 static const argument_t argument_mapping[] =428 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()464 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()