| /external/tensorflow/tensorflow/core/kernels/ |
| D | control_flow_ops.h | 27 class ControlTriggerOp : public OpKernel {
30 : OpKernel(context) {} in ControlTriggerOp()
39 class SwitchOp : public OpKernel {
41 explicit SwitchOp(OpKernelConstruction* context) : OpKernel(context) {} in SwitchOp()
52 class MergeOp : public OpKernel {
65 class EnterOp : public OpKernel {
67 explicit EnterOp(OpKernelConstruction* context) : OpKernel(context) {} in EnterOp()
78 class ExitOp : public OpKernel {
80 explicit ExitOp(OpKernelConstruction* context) : OpKernel(context) {} in ExitOp()
90 class NextIterationOp : public OpKernel {
[all …]
|
| D | summary_kernels.cc | 31 class CreateSummaryFileWriterOp : public OpKernel {
34 : OpKernel(ctx) {} in CreateSummaryFileWriterOp()
62 class CreateSummaryDbWriterOp : public OpKernel {
64 explicit CreateSummaryDbWriterOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in CreateSummaryDbWriterOp()
99 class FlushSummaryWriterOp : public OpKernel {
101 explicit FlushSummaryWriterOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in FlushSummaryWriterOp()
113 class CloseSummaryWriterOp : public OpKernel {
115 explicit CloseSummaryWriterOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in CloseSummaryWriterOp()
125 class WriteSummaryOp : public OpKernel {
127 explicit WriteSummaryOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in WriteSummaryOp()
[all …]
|
| D | save_op.cc | 29 class SaveOp : public OpKernel {
31 explicit SaveOp(OpKernelConstruction* context) : OpKernel(context) {} in SaveOp()
40 class SaveSlicesOp : public OpKernel {
42 explicit SaveSlicesOp(OpKernelConstruction* context) : OpKernel(context) {} in SaveSlicesOp()
51 class ShardedFilenameOp : public OpKernel {
53 explicit ShardedFilenameOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShardedFilenameOp()
74 class ShardedFilespecOp : public OpKernel {
76 explicit ShardedFilespecOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShardedFilespecOp()
|
| D | shape_ops.h | 40 class ShapeOp : public OpKernel {
42 explicit ShapeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShapeOp()
67 class ShapeNOp : public OpKernel {
69 explicit ShapeNOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShapeNOp()
96 class RankOp : public OpKernel {
98 explicit RankOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in RankOp()
113 class SizeOp : public OpKernel {
115 explicit SizeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in SizeOp()
136 class ExpandDimsOp : public OpKernel {
138 explicit ExpandDimsOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ExpandDimsOp()
[all …]
|
| D | base64_ops.cc | 27 class EncodeBase64Op : public OpKernel {
29 explicit EncodeBase64Op(OpKernelConstruction* context) : OpKernel(context) { in EncodeBase64Op()
54 class DecodeBase64Op : public OpKernel {
56 using OpKernel::OpKernel;
|
| D | stage_op.cc | 201 class StageOp : public OpKernel {
203 explicit StageOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in StageOp()
226 class UnstageOp : public OpKernel {
228 explicit UnstageOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in UnstageOp()
259 class StagePeekOp : public OpKernel {
261 explicit StagePeekOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in StagePeekOp()
296 class StageSizeOp : public OpKernel {
298 explicit StageSizeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in StageSizeOp()
326 class StageClearOp : public OpKernel {
328 explicit StageClearOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in StageClearOp()
|
| D | parse_tensor_op.cc | 28 class ParseTensorOp : public OpKernel {
30 explicit ParseTensorOp(OpKernelConstruction* context) : OpKernel(context) { in ParseTensorOp()
70 class SerializeTensorOp : public OpKernel {
72 using OpKernel::OpKernel;
|
| D | logging_ops.cc | 70 class AssertOp : public OpKernel {
72 explicit AssertOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in AssertOp()
108 class PrintOp : public OpKernel {
111 : OpKernel(ctx), call_counter_(0) { in PrintOp()
147 class PrintV2Op : public OpKernel {
149 explicit PrintV2Op(OpKernelConstruction* ctx) : OpKernel(ctx) { in PrintV2Op()
227 class TimestampOp : public OpKernel {
229 explicit TimestampOp(OpKernelConstruction* context) : OpKernel(context) {} in TimestampOp()
|
| D | session_ops.cc | 38 class GetSessionHandleOp : public OpKernel {
41 : OpKernel(context) {} in GetSessionHandleOp()
106 class GetSessionTensorOp : public OpKernel {
109 : OpKernel(context) {} in GetSessionTensorOp()
149 class DeleteSessionTensorOp : public OpKernel {
152 : OpKernel(context) {} in DeleteSessionTensorOp()
|
| D | fake_quant_ops.cc | 34 using tensorflow::OpKernel;
174 class FakeQuantWithMinMaxVarsOp : public OpKernel {
177 : OpKernel::OpKernel(context) { in FakeQuantWithMinMaxVarsOp()
213 class FakeQuantWithMinMaxVarsGradientOp : public OpKernel {
216 : OpKernel::OpKernel(context) { in FakeQuantWithMinMaxVarsGradientOp()
303 class FakeQuantWithMinMaxVarsPerChannelOp : public OpKernel {
306 : OpKernel::OpKernel(context) { in FakeQuantWithMinMaxVarsPerChannelOp()
349 class FakeQuantWithMinMaxVarsPerChannelGradientOp : public OpKernel {
353 : OpKernel::OpKernel(context) { in FakeQuantWithMinMaxVarsPerChannelGradientOp()
|
| D | whole_file_read_ops.cc | 98 class ReadFileOp : public OpKernel {
100 using OpKernel::OpKernel;
120 class WriteFileOp : public OpKernel {
122 using OpKernel::OpKernel;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 31 class IteratorHandleOp : public OpKernel {
82 class AnonymousIteratorHandleOp : public OpKernel {
101 class MakeIteratorOp : public OpKernel {
103 explicit MakeIteratorOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in MakeIteratorOp()
138 class IteratorGetNextSyncOp : public OpKernel {
140 explicit IteratorGetNextSyncOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in IteratorGetNextSyncOp()
145 class IteratorToStringHandleOp : public OpKernel {
148 : OpKernel(ctx) {} in IteratorToStringHandleOp()
153 class IteratorFromStringHandleOp : public OpKernel {
|
| D | optional_ops.h | 155 class OptionalNoneOp : public OpKernel {
157 explicit OptionalNoneOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in OptionalNoneOp()
162 class OptionalFromValueOp : public OpKernel {
164 explicit OptionalFromValueOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in OptionalFromValueOp()
169 class OptionalHasValueOp : public OpKernel {
171 explicit OptionalHasValueOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in OptionalHasValueOp()
176 class OptionalGetValueOp : public OpKernel {
178 explicit OptionalGetValueOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in OptionalGetValueOp()
|
| D | dataset_test_base.h | 65 std::unique_ptr<OpKernel>* op_kernel);
68 Status CreateDataset(OpKernel* kernel, OpKernelContext* context,
75 StringPiece node_name, std::unique_ptr<OpKernel>* range_op_kernel) { in CreateRangeDatasetOpKernel()
92 std::unique_ptr<OpKernel> range_kernel; in CreateRangeDataset()
115 std::unique_ptr<OpKernel>* tensor_slice_dataset_kernel);
137 Status RunOpKernel(OpKernel* op_kernel, OpKernelContext* context);
140 Status CheckOpKernelInput(const OpKernel& kernel,
144 Status CreateOpKernelContext(OpKernel* kernel,
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | resource_ops.cc | 25 class TensorForestCreateTreeVariableOp : public OpKernel {
28 : OpKernel(context) {} in TensorForestCreateTreeVariableOp()
52 class TensorForestTreeSerializeOp : public OpKernel {
55 : OpKernel(context) {} in TensorForestTreeSerializeOp()
72 class TensorForestTreeDeserializeOp : public OpKernel {
75 : OpKernel(context) {} in TensorForestTreeDeserializeOp()
99 class TensorForestTreeSizeOp : public OpKernel {
102 : OpKernel(context) {} in TensorForestTreeSizeOp()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | model_ops.cc | 32 class CreateTreeEnsembleVariableOp : public OpKernel {
35 : OpKernel(context) {} in CreateTreeEnsembleVariableOp()
66 class TreeEnsembleStampTokenOp : public OpKernel {
69 : OpKernel(context) {} in TreeEnsembleStampTokenOp()
85 class TreeEnsembleSerializeOp : public OpKernel {
88 : OpKernel(context) {} in TreeEnsembleSerializeOp()
109 class TreeEnsembleDeserializeOp : public OpKernel {
112 : OpKernel(context) {} in TreeEnsembleDeserializeOp()
140 class TreeEnsembleUsedHandlersOp : public OpKernel {
143 : OpKernel(context) { in TreeEnsembleUsedHandlersOp()
|
| D | stats_accumulator_ops.cc | 355 class CreateStatsAccumulatorScalarOp : public OpKernel {
358 : OpKernel(context) {} in CreateStatsAccumulatorScalarOp()
382 class CreateStatsAccumulatorTensorOp : public OpKernel {
385 : OpKernel(context) {} in CreateStatsAccumulatorTensorOp()
416 class StatsAccumulatorScalarAddOp : public OpKernel {
419 : OpKernel(context) {} in StatsAccumulatorScalarAddOp()
481 class StatsAccumulatorTensorAddOp : public OpKernel {
484 : OpKernel(context) {} in StatsAccumulatorTensorAddOp()
546 class StatsAccumulatorScalarFlushOp : public OpKernel {
549 : OpKernel(context) {} in StatsAccumulatorScalarFlushOp()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 89 class KernelLabelOp : public OpKernel {
91 using OpKernel::OpKernel;
130 class GraphDefVersionOp : public OpKernel {
133 : OpKernel(ctx), graph_def_version_(ctx->graph_def_version()) {} in GraphDefVersionOp()
148 class OldOp : public OpKernel {
150 explicit OldOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in OldOp()
168 class ResourceCreateOp : public OpKernel {
170 explicit ResourceCreateOp(OpKernelConstruction* c) : OpKernel(c) {} in ResourceCreateOp()
182 class ResourceUsingOp : public OpKernel {
184 explicit ResourceUsingOp(OpKernelConstruction* context) : OpKernel(context) {} in ResourceUsingOp()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 39 class DummyKernel : public tensorflow::OpKernel {
42 : OpKernel(context) {} in DummyKernel()
55 class TestOp2 : public ::tensorflow::OpKernel {
58 : OpKernel(context) { in TestOp2()
80 class TestOp3Cpu : public tensorflow::OpKernel {
82 explicit TestOp3Cpu(OpKernelConstruction* context) : OpKernel(context) {} in TestOp3Cpu()
91 class TestOp3Gpu : public tensorflow::OpKernel {
93 explicit TestOp3Gpu(OpKernelConstruction* context) : OpKernel(context) {} in TestOp3Gpu()
108 class TestOp5Cpu : public tensorflow::OpKernel {
110 explicit TestOp5Cpu(OpKernelConstruction* context) : OpKernel(context) {} in TestOp5Cpu()
[all …]
|
| D | op_segment_test.cc | 54 void ValidateOpAndTypes(OpKernel* op, const NodeDef& expected, DataType dt) { in ValidateOpAndTypes()
65 return [this, ndef](OpKernel** kernel) { in GetFn()
79 OpKernel* op; in TEST_F()
95 auto reterr = [](OpKernel** kernel) { in TEST_F()
116 OpKernel* op; in TEST_F()
124 OpKernel* op; in TEST_F()
135 OpKernel* op; in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | resource_ops.cc | 35 class BoostedTreesCreateEnsembleOp : public OpKernel {
38 : OpKernel(context) {} in BoostedTreesCreateEnsembleOp()
74 class BoostedTreesGetEnsembleStatesOp : public OpKernel {
77 : OpKernel(context) {} in BoostedTreesGetEnsembleStatesOp()
138 class BoostedTreesSerializeEnsembleOp : public OpKernel {
141 : OpKernel(context) {} in BoostedTreesSerializeEnsembleOp()
166 class BoostedTreesDeserializeEnsembleOp : public OpKernel {
169 : OpKernel(context) {} in BoostedTreesDeserializeEnsembleOp()
|
| D | quantile_ops.cc | 95 class BoostedTreesCreateQuantileStreamResourceOp : public OpKernel {
99 : OpKernel(context) { in BoostedTreesCreateQuantileStreamResourceOp()
137 class BoostedTreesMakeQuantileSummariesOp : public OpKernel {
141 : OpKernel(context) { in BoostedTreesMakeQuantileSummariesOp()
217 class BoostedTreesQuantileStreamResourceAddSummariesOp : public OpKernel {
221 : OpKernel(context) {} in BoostedTreesQuantileStreamResourceAddSummariesOp()
275 class BoostedTreesQuantileStreamResourceDeserializeOp : public OpKernel {
279 : OpKernel(context) { in BoostedTreesQuantileStreamResourceDeserializeOp()
326 class BoostedTreesQuantileStreamResourceFlushOp : public OpKernel {
330 : OpKernel(context) { in BoostedTreesQuantileStreamResourceFlushOp()
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | model_ops.cc | 36 class CreateTreeVariableOp : public OpKernel {
39 : OpKernel(context) { in CreateTreeVariableOp()
74 class TreeSerializeOp : public OpKernel {
76 explicit TreeSerializeOp(OpKernelConstruction* context) : OpKernel(context) {} in TreeSerializeOp()
93 class TreeDeserializeOp : public OpKernel {
96 : OpKernel(context) { in TreeDeserializeOp()
129 class TreeSizeOp : public OpKernel {
131 explicit TreeSizeOp(OpKernelConstruction* context) : OpKernel(context) {} in TreeSizeOp()
161 class TreePredictionsV4Op : public OpKernel {
164 : OpKernel(context) { in TreePredictionsV4Op()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.h | 34 class XlaCompileOnDemandOp : public OpKernel {
36 explicit XlaCompileOnDemandOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in XlaCompileOnDemandOp()
41 Status ShouldArgumentBeConstant(const OpKernel* op_kernel, int64 argument_idx,
43 Status MustArgumentBeConstant(const OpKernel* op_kernel, int64 argument_idx,
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 163 class XRTAllocateOp : public OpKernel {
165 explicit XRTAllocateOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in XRTAllocateOp()
213 class XRTAllocateFromTensorOp : public OpKernel {
215 explicit XRTAllocateFromTensorOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in XRTAllocateFromTensorOp()
317 class XRTSubTupleOp : public OpKernel {
319 explicit XRTSubTupleOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in XRTSubTupleOp()
373 class XRTMakeTupleOp : public OpKernel {
375 explicit XRTMakeTupleOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in XRTMakeTupleOp()
457 class XRTReadLiteralOp : public OpKernel {
459 explicit XRTReadLiteralOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in XRTReadLiteralOp()
[all …]
|