Home
last modified time | relevance | path

Searched refs:kFunc (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op.cc34 /* static */ constexpr const char* const MapDatasetOp::kFunc; member in tensorflow::data::MapDatasetOp
125 {std::make_pair(kFunc, f_attr), in AsGraphDefInternal()
220 FunctionMetadata::Create(ctx, kFunc, params, &func_metadata_)); in MapDatasetOp()
Dflat_map_dataset_op.cc34 /* static */ constexpr const char* const FlatMapDatasetOp::kFunc; member in tensorflow::data::FlatMapDatasetOp
106 {std::make_pair(kFunc, f), in AsGraphDefInternal()
277 OP_REQUIRES_OK(ctx, FunctionMetadata::Create(ctx, kFunc, /*params=*/{}, in FlatMapDatasetOp()
Dflat_map_dataset_op.h29 static constexpr const char* const kFunc = "f"; variable
Dmap_dataset_op.h29 static constexpr const char* const kFunc = "f"; variable
Dinterleave_dataset_op.h31 static constexpr const char* const kFunc = "f"; variable
Dparallel_map_dataset_op.h31 static constexpr const char* const kFunc = "f"; variable
Dparallel_interleave_dataset_op.h37 static constexpr const char* const kFunc = "f"; variable
Dmap_defun_op.h43 static constexpr const char* const kFunc = "f"; variable
Dinterleave_dataset_op.cc39 /* static */ constexpr const char* const InterleaveDatasetOp::kFunc; member in tensorflow::data::InterleaveDatasetOp
122 {{kFunc, f}, {kTarguments, other_arguments_types_attr}}, output)); in AsGraphDefInternal()
314 OP_REQUIRES_OK(ctx, FunctionMetadata::Create(ctx, kFunc, /*params=*/{}, in InterleaveDatasetOp()
Dmap_defun_op.cc38 /* static */ constexpr const char* const MapDefunOp::kFunc; member in tensorflow::data::MapDefunOp
172 OP_REQUIRES_OK(ctx, ctx->GetAttr(kFunc, &func)); in MapDefunOp()
Dparallel_map_dataset_op.cc47 /* static */ constexpr const char* const ParallelMapDatasetOp::kFunc; member in tensorflow::data::ParallelMapDatasetOp
161 attrs.emplace_back(kFunc, f_attr); in AsGraphDefInternal()
681 FunctionMetadata::Create(ctx, kFunc, params, &func_metadata_)); in ParallelMapDatasetOp()
Dflat_map_dataset_op_test.cc59 *attr_vector = {{FlatMapDatasetOp::kFunc, func_}, in GetAttributes()
Dmap_defun_op_test.cc71 {MapDefunOp::kFunc, func_}, in GetAttributes()
Dparallel_map_dataset_op_test.cc73 {ParallelMapDatasetOp::kFunc, func_}, in GetAttributes()
Dinterleave_dataset_op_test.cc72 *attr_vector = {{InterleaveDatasetOp::kFunc, func_}, in GetAttributes()
Dparallel_interleave_dataset_op.cc71 /* static */ constexpr const char* const ParallelInterleaveDatasetOp::kFunc; member in tensorflow::data::ParallelInterleaveDatasetOp
286 attrs.emplace_back(kFunc, f); in AsGraphDefInternal()
1574 OP_REQUIRES_OK(ctx, FunctionMetadata::Create(ctx, kFunc, /*params=*/{}, in ParallelInterleaveDatasetOp()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.h36 static constexpr const char* const kFunc = "f"; variable
Dparallel_interleave_dataset_op.h42 static constexpr const char* const kFunc = "f"; variable
Dmap_and_batch_dataset_op.cc54 /* static */ constexpr const char* const MapAndBatchDatasetOp::kFunc; member in tensorflow::data::experimental::MapAndBatchDatasetOp
178 {std::make_pair(kFunc, f), in AsGraphDefInternal()
693 OP_REQUIRES_OK(ctx, FunctionMetadata::Create(ctx, kFunc, /*params=*/{}, in MapAndBatchDatasetOp()
Dmap_and_batch_dataset_op_test.cc75 {MapAndBatchDatasetOp::kFunc, func_}, in GetAttributes()
Dparallel_interleave_dataset_op.cc58 /* static */ constexpr const char* const ParallelInterleaveDatasetOp::kFunc; member in tensorflow::data::experimental::ParallelInterleaveDatasetOp
207 attrs.emplace_back(kFunc, f); in AsGraphDefInternal()
1176 OP_REQUIRES_OK(ctx, FunctionMetadata::Create(ctx, kFunc, /*params=*/{}, in ParallelInterleaveDatasetOp()
Dparallel_interleave_dataset_op_test.cc86 {ParallelInterleaveDatasetOp::kFunc, func_}, in GetAttributes()
/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.cc288 case AttrValue::kFunc: { in SummarizeAttrValue()
330 VALIDATE_FIELD(func, "func", kFunc); in AttrValueHasType()
616 case AttrValue::kFunc: in HasPlaceHolder()
644 case AttrValue::kFunc: in SubstitutePlaceholders()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_attr.cc94 case AttrValue::kFunc: { in ConvertAttributeValue()
/external/tensorflow/tensorflow/lite/toco/
Dtensorflow_util.cc75 if (attr.second.value_case() == AttrValue::kFunc) { in LogDumpGraphDef()

12