/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 67 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::CreateSummaryDbWriterOp 103 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::FlushSummaryWriterOp 116 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::CloseSummaryWriterOp 128 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteSummaryOp 152 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteRawProtoSummaryOp 188 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ImportEventOp 208 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteScalarSummaryOp 230 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::WriteHistogramSummaryOp 258 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 | list_kernels.cc | 135 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::EmptyTensorList 180 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListPushBack 234 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListLength 258 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListElementShape 301 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListReserve 340 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListResize 404 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListSetItem 459 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::TensorListConcatLists
|
D | constant_op.cc | 86 void ConstantOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ConstantOp 146 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::FillOp 239 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ZerosLikeOp 305 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::OnesLikeOp 350 void PlaceholderOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::PlaceholderOp
|
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 | logging_ops.cc | 57 void AssertOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::AssertOp 90 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::PrintOp 143 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::PrintV2Op 213 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::TimestampOp
|
D | searchsorted_op.cc | 35 static Status Compute(OpKernelContext* context, in Compute() function 58 static Status Compute(OpKernelContext* context, in Compute() function 85 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::UpperBoundOp 130 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::LowerBoundOp
|
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 133 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::DeleteSessionTensorOp
|
D | list_kernels.h | 68 void Compute(OpKernelContext* c) override { in Compute() function 168 void Compute(OpKernelContext* c) override { in Compute() function 233 void Compute(OpKernelContext* c) override { in Compute() function 289 void Compute(OpKernelContext* c) override { in Compute() function 477 void Compute(OpKernelContext* c) override { in Compute() function 556 void Compute(OpKernelContext* c) override { in Compute() function 652 void Compute(OpKernelContext* c) override { in Compute() function 722 void Compute(OpKernelContext* c) override { in Compute() function 772 void Compute(OpKernelContext* c) override { in Compute() function 893 void Compute(OpKernelContext* c) override { in Compute() function
|
D | cwise_ops_common.h | 88 void Compute(OpKernelContext* ctx) override { in Compute() function 230 void Compute(OpKernelContext* context) override { in Compute() function 265 void Compute(OpKernelContext* ctx) override { in Compute() function 306 void Compute(OpKernelContext* ctx) override { in Compute() function 325 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 | resource_variable_ops.cc | 133 void ReadVariableOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ReadVariableOp 172 void ReadVariablesOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ReadVariablesOp 246 void VarHandleOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::VarHandleOp 354 void DestroyResourceOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::DestroyResourceOp 381 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::AssignVariableOp 452 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::AssignVariableOp 545 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::AssignUpdateVariableOp 608 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::VarIsInitializedOp 648 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::ResourceGatherOp 826 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::ResourceGatherNdOp [all …]
|
/external/tensorflow/tensorflow/python/framework/ |
D | test_ops.cc | 101 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::KernelLabelOp 143 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::GraphDefVersionOp 160 void Compute(OpKernelContext* ctx) override {} in Compute() function in tensorflow::OldOp 180 void Compute(OpKernelContext* c) override { in Compute() function in tensorflow::ResourceCreateOp 194 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::ResourceUsingOp 208 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::TestAttrOp 683 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::DevicePlacementOp 709 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::DTypeWithDefaultOp 733 void Compute(OpKernelContext* ctx) override { in Compute() function in tensorflow::IsTensorFloat32Enabled
|
/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_configuration_ops.cc | 125 void ConfigureDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ConfigureDistributedTpuOp 161 void WaitForDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::WaitForDistributedTpuOp 251 void ShutdownDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::ShutdownDistributedTpuOp 267 void InitializeHostForDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::InitializeHostForDistributedTpuOp 386 void SetGlobalTPUArrayOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::SetGlobalTPUArrayOp 402 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 495 void Compute(OpKernelContext* ctx) override { in Compute() function 546 void Compute(OpKernelContext* ctx) override { in Compute() function 630 void Compute(OpKernelContext* ctx) override { in Compute() function 681 void Compute(OpKernelContext* ctx) override { in Compute() function 710 void Compute(OpKernelContext* ctx) override { in Compute() function [all …]
|
/external/tensorflow/tensorflow/c/experimental/gradients/ |
D | math_grad.cc | 53 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::AddGradientFunction 74 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::ExpGradientFunction 98 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::SqrtGradientFunction 124 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::MatMulGradientFunction 220 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::NegGradientFunction 239 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::SubGradientFunction 276 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::MulGradientFunction 323 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::Log1pGradientFunction 394 Status Compute(AbstractContext* ctx, in Compute() function in tensorflow::gradients::__anond651015b0111::DivNoNanGradientFunction
|
/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 | 81 void DatasetToGraphOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::DatasetToGraphOp 134 void DatasetCardinalityOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::DatasetCardinalityOp 142 void DatasetFromGraphOp::Compute(OpKernelContext* ctx) { in Compute() function in tensorflow::data::DatasetFromGraphOp
|
/external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
D | resource_ops.cc | 41 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::BoostedTreesCreateEnsembleOp 81 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::BoostedTreesGetEnsembleStatesOp 144 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::BoostedTreesSerializeEnsembleOp 171 void Compute(OpKernelContext* context) override { in Compute() function in tensorflow::BoostedTreesDeserializeEnsembleOp
|
/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
|