Searched refs:argument_t (Results 1 – 11 of 11) sorted by relevance
37 const Tensor* argument_t; in ParseScalarArgument() local38 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseScalarArgument()39 if (!TensorShapeUtils::IsScalar(argument_t->shape())) { in ParseScalarArgument()42 *output = argument_t->scalar<T>()(); in ParseScalarArgument()
91 const Tensor* argument_t; in ParseScalarArgument() local92 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseScalarArgument()93 if (!TensorShapeUtils::IsScalar(argument_t->shape())) { in ParseScalarArgument()96 *output = argument_t->scalar<T>()(); in ParseScalarArgument()
68 const Tensor* argument_t; in ParseScalarArgument() local69 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseScalarArgument()70 if (!TensorShapeUtils::IsScalar(argument_t->shape())) { in ParseScalarArgument()73 *output = argument_t->scalar<T>()(); in ParseScalarArgument()
859 const Tensor* argument_t; in ParseScalarArgument() local860 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseScalarArgument()861 if (!TensorShapeUtils::IsScalar(argument_t->shape())) { in ParseScalarArgument()864 *output = argument_t->scalar<T>()(); in ParseScalarArgument()872 const Tensor* argument_t; in ParseVectorArgument() local873 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseVectorArgument()874 if (!TensorShapeUtils::IsVector(argument_t->shape())) { in ParseVectorArgument()877 int size = argument_t->vec<T>().size(); in ParseVectorArgument()880 output->push_back(argument_t->vec<T>()(i)); in ParseVectorArgument()
343 const Tensor* argument_t; in ParseScalarArgument() local344 TF_RETURN_IF_ERROR(ctx->input(argument_name, &argument_t)); in ParseScalarArgument()345 if (!TensorShapeUtils::IsScalar(argument_t->shape())) { in ParseScalarArgument()348 *output = argument_t->scalar<T>()(); in ParseScalarArgument()
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()
285 }argument_t; typedef287 static const argument_t argument_mapping[] =1075 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()1112 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()
286 } argument_t; typedef288 static const argument_t argument_mapping[] =1083 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()1120 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()
137 } argument_t; typedef139 static const argument_t argument_mapping[] =426 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in print_usage()462 WORD32 num_entries = sizeof(argument_mapping) / sizeof(argument_t); in get_argument()
284 }argument_t; typedef286 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()