| /external/tensorflow/tensorflow/core/kernels/ |
| D | aggregate_ops.h | 37 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute() function
54 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute() function
73 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute() function
94 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute() function
117 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute() function
142 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute() function
166 static void Compute( in Compute() function
190 static void Compute( in Compute() function
213 static void Compute( in Compute() function
|
| D | summary_kernels.cc | 38 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::CreateSummaryFileWriterOp
75 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::CreateSummaryDbWriterOp
111 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::FlushSummaryWriterOp
124 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::CloseSummaryWriterOp
136 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteSummaryOp
160 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteRawProtoSummaryOp
196 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ImportEventOp
216 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteScalarSummaryOp
238 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteHistogramSummaryOp
266 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteImageSummaryOp
[all …]
|
| D | control_flow_ops.cc | 26 void SwitchOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::SwitchOp
42 void SwitchNOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::SwitchNOp
166 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::RefSelectOp
208 void MergeOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::MergeOp
295 void EnterOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::EnterOp
352 void ExitOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::ExitOp
404 void NextIterationOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::NextIterationOp
460 void LoopCondOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::LoopCondOp
501 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::AbortOp
|
| D | save_op.cc | 33 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::SaveOp
44 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::SaveSlicesOp
55 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ShardedFilenameOp
78 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ShardedFilespecOp
|
| D | map_kernels.h | 83 void Compute(OpKernelContext* ctx) override { in Compute() function
98 void Compute(OpKernelContext* ctx) override { in Compute() function
112 void Compute(OpKernelContext* ctx) override { in Compute() function
132 void Compute(OpKernelContext* ctx) override { in Compute() function
149 void Compute(OpKernelContext* ctx) override { in Compute() function
172 void Compute(OpKernelContext* ctx) override { in Compute() function
189 void Compute(OpKernelContext* ctx) override { in Compute() function
|
| D | shape_ops.h | 46 void Compute(OpKernelContext* ctx) override { in Compute() function
73 void Compute(OpKernelContext* ctx) override { in Compute() function
102 void Compute(OpKernelContext* ctx) override { in Compute() function
119 void Compute(OpKernelContext* ctx) override { in Compute() function
142 void Compute(OpKernelContext* ctx) override { in Compute() function
201 void Compute(OpKernelContext* ctx) override { in Compute() function
|
| D | constant_op.cc | 90 void ConstantOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ConstantOp
150 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::FillOp
243 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ZerosLikeOp
308 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::OnesLikeOp
352 void PlaceholderOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::PlaceholderOp
|
| D | list_kernels.cc | 140 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::EmptyTensorList
191 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListPushBack
248 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListLength
276 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListElementShape
324 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListReserve
375 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListResize
443 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListSetItem
511 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListConcatLists
|
| D | fact_op.cc | 83 void Compute(OpKernelContext* context, const char* const facts[], in Compute() function in tensorflow::FactOpKernel
101 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::FactOpKernel1
111 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::FactOpKernel2
|
| D | session_ops.cc | 45 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::GetSessionHandleOp
99 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::GetSessionTensorOp
135 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::DeleteSessionTensorOp
|
| D | logging_ops.cc | 58 void AssertOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::AssertOp
91 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::PrintOp
144 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::PrintV2Op
214 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::TimestampOp
|
| D | cwise_ops_common.h | 88 void Compute(OpKernelContext* ctx) override { in Compute() function
240 void Compute(OpKernelContext* context) override { in Compute() function
275 void Compute(OpKernelContext* ctx) override { in Compute() function
316 void Compute(OpKernelContext* ctx) override { in Compute() function
335 void Compute(OpKernelContext* ctx) override { in Compute() function
|
| D | restore_op.cc | 42 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::RestoreOp
68 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::RestoreSliceOp
|
| D | variable_ops.cc | 75 void VariableOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::VariableOp
106 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::TemporaryVariableOp
154 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::DestroyTemporaryVariableOp
182 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::IsVariableInitializedOp
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_ops.cc | 131 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::KernelLabelOp
171 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::GraphDefVersionOp
188 void Compute(OpKernelContext* ctx) override {} in Compute() function in tensorflow::OldOp
197 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::GetDeadlineOp
213 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::SleepOp
224 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::SleepIdentityOp
248 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::ResourceCreateOp
262 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ResourceUsingOp
277 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::IsResourceHandleRefCountingOp
293 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::MakeWeakResourceHandleOp
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op.cc | 35 void TpuCompileOp::Compute(OpKernelContext* ctx) { impl_->Compute(ctx); } in Compute() function in tensorflow::tpu::TpuCompileOp
44 void TpuCompileMlirOp::Compute(OpKernelContext* ctx) { impl_->Compute(ctx); } in Compute() function in tensorflow::tpu::TpuCompileMlirOp
46 void TpuCompileSucceededAssertOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::tpu::TpuCompileSucceededAssertOp
|
| D | tpu_embedding_configuration_ops.cc | 57 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ExecuteTPUEmbeddingPartitionerOp
108 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ConfigureTPUEmbeddingMemoryOp
161 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::CollateTPUEmbeddingMemoryOp
230 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ConfigureTPUEmbeddingHostOp
299 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ConnectTPUEmbeddingHostsOp
341 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::FinalizeTPUEmbeddingOp
389 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::IsTPUEmbeddingInitializedOp
|
| D | tpu_configuration_ops.cc | 131 void ConfigureDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ConfigureDistributedTpuOp
167 void WaitForDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::WaitForDistributedTpuOp
257 void ShutdownDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ShutdownDistributedTpuOp
273 void InitializeHostForDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::InitializeHostForDistributedTpuOp
443 void SetGlobalTPUArrayOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::SetGlobalTPUArrayOp
459 void DisconnectDistributedTpuChipsOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::DisconnectDistributedTpuChipsOp
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 174 void Compute(OpKernelContext* ctx) override { in Compute() function
229 void Compute(OpKernelContext* ctx) override { in Compute() function
302 void Compute(OpKernelContext* ctx) override { in Compute() function
373 void Compute(OpKernelContext* ctx) override { in Compute() function
424 void Compute(OpKernelContext* ctx) override { in Compute() function
496 void Compute(OpKernelContext* ctx) override { in Compute() function
547 void Compute(OpKernelContext* ctx) override { in Compute() function
631 void Compute(OpKernelContext* ctx) override { in Compute() function
682 void Compute(OpKernelContext* ctx) override { in Compute() function
711 void Compute(OpKernelContext* ctx) override { in Compute() function
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cc | 69 void OptionalNoneOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::OptionalNoneOp
73 void OptionalFromValueOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::OptionalFromValueOp
82 void OptionalHasValueOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::OptionalHasValueOp
100 void OptionalGetValueOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::OptionalGetValueOp
|
| D | dataset_ops.cc | 82 void DatasetToGraphOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::DatasetToGraphOp
135 void DatasetCardinalityOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::DatasetCardinalityOp
143 void DatasetFromGraphOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::DatasetFromGraphOp
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | math_grad.cc | 53 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::AddGradientFunction
74 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::ExpGradientFunction
98 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::SqrtGradientFunction
124 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::MatMulGradientFunction
220 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::NegGradientFunction
239 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::SubGradientFunction
276 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::MulGradientFunction
323 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::Log1pGradientFunction
394 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anon217bc5340111::DivNoNanGradientFunction
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.cc | 27 void XlaDeviceDummyOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::XlaDeviceDummyOp
37 void XlaAssignVariableOp::Compute(OpKernelContext* context) { in Compute() function in tensorflow::XlaAssignVariableOp
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | light_outside_compilation_kernels_for_test.cc | 41 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::__anon8a0b77420111::TestStaticTfOp
78 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::__anon8a0b77420111::TestStaticMultipleOutputTfOp
126 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::__anon8a0b77420111::TestDynamicTfOp
190 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::__anon8a0b77420111::DynamicMultidimOp
263 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::__anon8a0b77420111::TestTfMustBeConstantOp
320 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::__anon8a0b77420111::TestDynamicTfWithBoundOp
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/ |
| D | simple_hash_table_kernel.cc | 139 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::custom_op_examples::SimpleHashTableCreateOpKernel
175 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::custom_op_examples::SimpleHashTableFindOpKernel
198 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::custom_op_examples::SimpleHashTableInsertOpKernel
217 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::custom_op_examples::SimpleHashTableRemoveOpKernel
234 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::custom_op_examples::SimpleHashTableExportOpKernel
251 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::custom_op_examples::SimpleHashTableImportOpKernel
|