| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op.h | 38 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate()
49 static void ValidateSameSizeHelper(OpKernelContext* context, const Tensor& g, in ValidateSameSizeHelper()
54 static bool ValidateSameSize(OpKernelContext* context, const Tensor& g, in ValidateSameSize()
66 void OperateNoTemplate(OpKernelContext* context, const Tensor& g,
76 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate()
83 void ReluGradOp<Device, T>::OperateNoTemplate(OpKernelContext* context, in OperateNoTemplate()
97 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate()
109 void OperateNoTemplate(OpKernelContext* context, const Tensor& g,
118 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate()
125 void Relu6GradOp<Device, T>::OperateNoTemplate(OpKernelContext* context, in OperateNoTemplate()
[all …]
|
| D | queue_op.h | 38 void Compute(OpKernelContext* context) override;
70 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final;
73 virtual void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
99 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
124 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
145 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
170 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
213 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
231 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
250 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
[all …]
|
| D | meta_support.h | 29 class OpKernelContext; variable
70 void QuantizedGemm(OpKernelContext* context, bool transpose_a, bool transpose_b,
79 void Requantize(OpKernelContext* context, const qint32* input, int count,
85 void Dequantize(OpKernelContext* context, const quint8* input, int count,
90 void Quantize(OpKernelContext*, const float* input, int count, float range_min,
98 void QuantizedBiasAdd(OpKernelContext* context, const quint8* input,
106 void Clamp(OpKernelContext* context, const quint8* input, int input_count,
|
| D | control_flow_ops.h | 31 void Compute(OpKernelContext* context) override {} in Compute()
42 void Compute(OpKernelContext* context) override;
57 void Compute(OpKernelContext* context) override;
70 void Compute(OpKernelContext* context) override;
83 void Compute(OpKernelContext* context) override;
96 void Compute(OpKernelContext* context) override;
108 void Compute(OpKernelContext* context) override;
124 void Compute(OpKernelContext* context) override;
|
| D | conditional_accumulator_base_op.h | 58 void Compute(OpKernelContext* ctx) override { in Compute()
77 virtual void SetHandleToOutput(OpKernelContext* ctx)
80 virtual Status CheckSignature(OpKernelContext* ctx) = 0;
98 Status SetAccumulatorHandle(OpKernelContext* ctx) in SetAccumulatorHandle()
134 void Compute(OpKernelContext* ctx) final { in Compute()
142 virtual void Compute(OpKernelContext* ctx,
148 virtual void CheckSignature(OpKernelContext* ctx, in CheckSignature()
168 void Compute(OpKernelContext* ctx, in Compute()
197 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
208 virtual void ComputeAsync(OpKernelContext* ctx,
[all …]
|
| D | transpose_op.h | 28 void Compute(OpKernelContext* ctx) override;
31 virtual Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
41 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
51 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
61 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
73 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
85 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
97 Status DoTranspose(OpKernelContext* ctx, const Tensor& in,
|
| D | fifo_queue.h | 42 void TryEnqueue(const Tuple& tuple, OpKernelContext* ctx,
44 void TryEnqueueMany(const Tuple& tuple, OpKernelContext* ctx,
46 void TryDequeue(OpKernelContext* ctx, CallbackWithTuple callback) override;
47 void TryDequeueMany(int num_elements, OpKernelContext* ctx,
61 void DequeueLocked(OpKernelContext* ctx, Tuple* tuple)
66 OpKernelContext* ctx,
|
| D | priority_queue.h | 59 void TryEnqueue(const Tuple& tuple, OpKernelContext* ctx,
61 void TryEnqueueMany(const Tuple& tuple, OpKernelContext* ctx,
63 void TryDequeue(OpKernelContext* ctx, CallbackWithTuple callback) override;
64 void TryDequeueMany(int num_elements, OpKernelContext* ctx,
80 void DequeueLocked(OpKernelContext* ctx, Tuple* tuple)
85 OpKernelContext* ctx,
|
| D | reader_ops.cc | 32 void Compute(OpKernelContext* context) override { in Compute()
41 virtual void ComputeWithReader(OpKernelContext* context,
56 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync()
69 virtual void ComputeWithReader(OpKernelContext* context,
80 void ComputeWithReader(OpKernelContext* context, in ComputeWithReader()
109 void ComputeWithReader(OpKernelContext* context, in ComputeWithReader()
163 void ComputeWithReader(OpKernelContext* context, in ComputeWithReader()
181 void ComputeWithReader(OpKernelContext* context, in ComputeWithReader()
200 void ComputeWithReader(OpKernelContext* context, in ComputeWithReader()
219 void ComputeWithReader(OpKernelContext* context, in ComputeWithReader()
[all …]
|
| D | lookup_util.h | 36 Status GetLookupTable(StringPiece input_name, OpKernelContext* ctx,
38 Status GetResourceLookupTable(StringPiece input_name, OpKernelContext* ctx,
40 Status GetReferenceLookupTable(StringPiece input_name, OpKernelContext* ctx,
46 Status GetInitializableLookupTable(StringPiece input_name, OpKernelContext* ctx,
61 void InitializeTableFromDataset(OpKernelContext* ctx,
|
| D | conditional_accumulator_base.h | 59 virtual void TryApplyGrad(int64 local_step, OpKernelContext* ctx) = 0;
60 void TryTakeGrad(int num_required, OpKernelContext* ctx,
83 virtual void DivideAccumGradByCounter(OpKernelContext* ctx)
85 virtual bool SetOutput(OpKernelContext* ctx) = 0;
95 OpKernelContext* context;
102 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
156 bool TakeGradLockedHelper(OpKernelContext* ctx, DoneCallback callback)
|
| D | save_restore_tensor.h | 24 class OpKernelContext; variable
36 OpKernelContext* context,
52 void RestoreTensor(OpKernelContext* context,
66 Status RestoreTensorsV2(OpKernelContext* context, const Tensor& prefix,
|
| D | map_kernels.h | 25 Status GetInputMap(OpKernelContext* ctx, int index, const TensorMap** ret_map) { in GetInputMap()
41 Status ForwardInputOrCreateNewMap(OpKernelContext* ctx, int32 input_index, in ForwardInputOrCreateNewMap()
83 void Compute(OpKernelContext* ctx) override { in Compute()
98 void Compute(OpKernelContext* ctx) override { in Compute()
112 void Compute(OpKernelContext* ctx) override { in Compute()
132 void Compute(OpKernelContext* ctx) override { in Compute()
149 void Compute(OpKernelContext* ctx) override { in Compute()
172 void Compute(OpKernelContext* ctx) override { in Compute()
189 void Compute(OpKernelContext* ctx) override { in Compute()
224 Status TensorMapBinaryAdd(OpKernelContext* ctx, const TensorMap& a, in TensorMapBinaryAdd()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels.h | 39 Status operator()(OpKernelContext* c, TTypes<int64>::ConstMatrix indices,
85 Status operator()(OpKernelContext* c,
102 Status operator()(OpKernelContext* c, const int rows, const int cols,
138 Status operator()(OpKernelContext* c,
148 Status Compute(OpKernelContext* ctx, const ConstCSRComponent<T>& a,
158 Status Compute(OpKernelContext* ctx, const ConstCSRComponent<T>& a,
188 explicit CSRSparseMatrixAdd(OpKernelContext* ctx, const T alpha,
196 explicit CSRSparseSparseMatrixMatMul(OpKernelContext* ctx, bool transpose_a,
203 Status operator()(OpKernelContext* ctx, const ConstCSRComponent<T>& x,
210 Status operator()(OpKernelContext* ctx, bool conjugate,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 51 Status GetNext(OpKernelContext* ctx, std::vector<Tensor>* out_tensors,
58 Status Restore(OpKernelContext* ctx, IteratorStateReader* reader);
65 Status SetIteratorFromDataset(OpKernelContext* ctx, DatasetBase* dataset);
149 void Compute(OpKernelContext* context) override TF_LOCKS_EXCLUDED(mu_);
160 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr,
184 Status CreateResource(OpKernelContext* ctx,
215 void Compute(OpKernelContext* ctx) final;
216 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) final;
219 virtual Status DoCompute(OpKernelContext* ctx) = 0;
231 Status DoCompute(OpKernelContext* ctx) override;
[all …]
|
| D | optional_ops.h | 31 Status WriteOptionalWithValueToOutput(OpKernelContext* ctx, int output_index,
36 Status WriteOptionalNoneToOutput(OpKernelContext* ctx, int output_index);
108 Status OptionalZerosLike(OpKernelContext* ctx, const OptionalVariant& x, in OptionalZerosLike()
125 Status OptionalBinaryAdd(OpKernelContext* ctx, const OptionalVariant& a, in OptionalBinaryAdd()
159 void Compute(OpKernelContext* ctx) override;
166 void Compute(OpKernelContext* ctx) override;
173 void Compute(OpKernelContext* ctx) override;
189 void Compute(OpKernelContext* ctx) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_util.h | 39 OpKernelContext::Params sub_params_;
46 std::unique_ptr<OpKernelContext> sub_ctx_;
47 SubContext(OpKernelContext* ctx, OpKernelContext::Params* params,
52 Status ComputeBinOp(OpKernelContext* op_ctx, OpKernelContext::Params* params,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 174 string OpKernel::ShapeTraceString(const OpKernelContext& ctx) const { in ShapeTraceString()
196 string OpKernel::TraceString(const OpKernelContext& ctx, bool verbose) const { in TraceString()
208 void AsyncOpKernel::Compute(OpKernelContext* context) { in Compute()
222 Tensor* PersistentTensor::AccessTensor(OpKernelContext* context) { in AccessTensor()
322 const int OpKernelContext::Params::kNeverForward;
323 const int OpKernelContext::Params::kNoReservation;
325 OpKernelContext::OpKernelContext(Params* params) in OpKernelContext() function in tensorflow::OpKernelContext
326 : OpKernelContext( in OpKernelContext()
329 OpKernelContext::OpKernelContext(Params* params, int num_outputs) in OpKernelContext() function in tensorflow::OpKernelContext
347 OpKernelContext::~OpKernelContext() { in ~OpKernelContext()
[all …]
|
| D | lookup_interface.h | 25 class OpKernelContext; variable
50 virtual Status Find(OpKernelContext* ctx, const Tensor& keys, Tensor* values,
64 virtual Status Insert(OpKernelContext* ctx, const Tensor& keys,
76 virtual Status Remove(OpKernelContext* ctx, const Tensor& keys) = 0;
85 virtual Status ExportValues(OpKernelContext* ctx) = 0;
91 virtual Status ImportValues(OpKernelContext* ctx, const Tensor& keys,
|
| D | queue_interface.h | 43 virtual void TryEnqueue(const Tuple& tuple, OpKernelContext* ctx,
48 virtual void TryEnqueueMany(const Tuple& tuple, OpKernelContext* ctx,
54 virtual void TryDequeue(OpKernelContext* ctx, CallbackWithTuple callback) = 0;
62 virtual void TryDequeueMany(int num_elements, OpKernelContext* ctx,
77 virtual void Close(OpKernelContext* ctx, bool cancel_pending_enqueues,
|
| D | resource_mgr.h | 305 OpKernelContext* ctx, const std::string& container, const std::string& name,
328 Status MakeResourceHandleToOutput(OpKernelContext* context, int output_index,
334 const ResourceHandle& HandleFromInput(OpKernelContext* ctx, int input);
335 Status HandleFromInput(OpKernelContext* ctx, StringPiece input,
343 Status CreateResource(OpKernelContext* ctx, const ResourceHandle& p, T* value);
350 Status LookupResource(OpKernelContext* ctx, const ResourceHandle& p, T** value);
357 Status LookupResource(OpKernelContext* ctx, const ResourceHandle& p,
363 Status LookupResources(OpKernelContext* ctx, absl::Span<ResourceHandle const> p,
376 Status LookupOrCreateResource(OpKernelContext* ctx, const ResourceHandle& p,
381 Status LookupOrCreateResource(OpKernelContext* ctx, const ResourceHandle& p,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | linalg_ops_common.h | 43 void Compute(OpKernelContext* context) override;
50 virtual int NumMatrixInputs(const OpKernelContext* context) const { in NumMatrixInputs()
58 OpKernelContext* context, const TensorShapes& input_matrix_shapes) const { in ValidateInputMatrixShapes()
65 static void ValidateSingleMatrix(OpKernelContext* context,
69 OpKernelContext* context, const TensorShapes& input_matrix_shapes);
71 static void ValidateSolver(OpKernelContext* context,
75 static void ValidateSquareSolver(OpKernelContext* context,
149 virtual void ComputeMatrix(OpKernelContext* context,
173 void ComputeTensorSlice(OpKernelContext* context, int64 matrix_index,
179 void AnalyzeInputs(OpKernelContext* context, TensorInputs* inputs,
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | transfer_ops.h | 34 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
37 virtual Status DoWork(OpKernelContext* context,
41 Status RunTransferWithOrdinal(OpKernelContext* ctx, int device_ordinal);
45 virtual Status RunTransfer(OpKernelContext* ctx) = 0;
64 Status RunTransfer(OpKernelContext* ctx) override;
80 Status RunTransfer(OpKernelContext* ctx) override;
|
| D | tpu_configuration_ops.h | 32 OpKernelContext* ctx);
47 void Compute(OpKernelContext* ctx) override;
71 void Compute(OpKernelContext* ctx) override;
91 void Compute(OpKernelContext* ctx) override;
114 void Compute(OpKernelContext* ctx) override;
135 void Compute(OpKernelContext* ctx) override;
154 void Compute(OpKernelContext* ctx) override;
|
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_solvers.h | 70 explicit ROCmSolver(OpKernelContext* context);
81 OpKernelContext* context() { return context_; }
89 OpKernelContext* context_; // not owned.
102 ScratchSpace(OpKernelContext* context, int64 size, bool on_host)
105 ScratchSpace(OpKernelContext* context, int64 size, const string& debug_info,
109 ScratchSpace(OpKernelContext* context, const TensorShape& shape,
148 OpKernelContext* context() const { return context_; }
151 OpKernelContext* context_; // not owned
|