| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_cross_op.cc | 326 OpInputList ragged_values_list; in Compute()
327 OpInputList ragged_splits_list; in Compute()
328 OpInputList sparse_indices_list; in Compute()
329 OpInputList sparse_values_list; in Compute()
330 OpInputList sparse_shape_list; in Compute()
331 OpInputList dense_list; in Compute()
378 Status ValidateInput(const OpInputList& ragged_values_list, in ValidateInput()
379 const OpInputList& ragged_splits_list, in ValidateInput()
380 const OpInputList& sparse_indices_list, in ValidateInput()
381 const OpInputList& sparse_values_list, in ValidateInput()
[all …]
|
| D | sparse_cross_op.cc | 464 int64_t CalculateBatchSize(const OpInputList& shapes_list_in, in CalculateBatchSize()
465 const OpInputList& dense_list_in) { in CalculateBatchSize()
478 Status ValidateInput(const OpInputList& indices_list_in, in ValidateInput()
479 const OpInputList& values_list_in, in ValidateInput()
480 const OpInputList& shapes_list_in, in ValidateInput()
481 const OpInputList& dense_list_in, in ValidateInput()
599 const OpInputList& indices_list_in, int64_t batch_size, in ExtractFeatureData()
640 GenerateColumnsFromInput(const OpInputList& indices_list_in, in GenerateColumnsFromInput()
641 const OpInputList& values_list_in, in GenerateColumnsFromInput()
642 const OpInputList& shapes_list_in, in GenerateColumnsFromInput()
[all …]
|
| D | sparse_concat_op.h | 27 void operator()(OpKernelContext* context, const OpInputList& inds,
28 const OpInputList& vals, const OpInputList& shapes,
|
| D | sparse_concat_op.cc | 43 void operator()(OpKernelContext* context, const OpInputList& inds, in operator ()()
44 const OpInputList& vals, const OpInputList& shapes, in operator ()()
100 OpInputList inds; in Compute()
110 OpInputList vals; in Compute()
122 OpInputList shapes; in Compute()
|
| D | sdca_internal.cc | 96 OpInputList sparse_indices_inputs; in Initialize()
99 OpInputList sparse_weights_inputs; in Initialize()
106 OpInputList dense_weights_inputs; in Initialize()
144 [&](const OpInputList& weight_inputs, OpOutputList* const weight_outputs, in Initialize()
341 OpInputList sparse_example_indices_inputs; in Initialize()
349 OpInputList sparse_feature_indices_inputs; in Initialize()
357 OpInputList sparse_feature_values_inputs; in Initialize()
389 OpInputList dense_features_inputs; in Initialize()
423 const OpInputList& sparse_example_indices_inputs, in CreateSparseFeatureRepresentation()
424 const OpInputList& sparse_feature_indices_inputs, in CreateSparseFeatureRepresentation()
[all …]
|
| D | quantized_concat_op.cc | 79 const OpInputList& input_mins, const OpInputList& input_maxes, in CalculateInputAndOutputRange()
118 int input_dims, const OpInputList& values, in CalculateConcatDims()
169 OpInputList values; in Compute()
172 OpInputList input_mins; in Compute()
178 OpInputList input_maxes; in Compute()
|
| D | dynamic_stitch_op.cc | 76 OpInputList* indices_inputs, in CheckArgsAndAllocateResult()
77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult()
165 OpInputList indices_inputs; in Compute()
166 OpInputList data_inputs; in Compute()
237 OpInputList indices_inputs; in Compute()
238 OpInputList data_inputs; in Compute()
|
| D | sdca_internal.h | 354 const OpInputList& sparse_example_indices_inputs,
355 const OpInputList& sparse_feature_indices_inputs,
356 const OpInputList& sparse_feature_values_inputs,
365 const OpInputList& dense_features_inputs,
|
| D | example_parsing_ops.cc | 61 OpInputList dense_defaults; in Compute()
116 OpInputList key_list; in GetInputListKeys()
127 const OpInputList& dense_defaults, in CheckInputShapes()
212 const OpInputList& dense_defaults) const { in MakeConfig()
306 OpInputList dense_defaults; in Compute()
409 OpInputList context_dense_defaults; in Compute()
489 const OpInputList& context_dense_defaults, in CheckInputShapes()
571 const OpInputList& context_dense_defaults) const { in MakeContextConfig()
787 OpInputList context_dense_keys; in Compute()
788 OpInputList context_sparse_keys; in Compute()
[all …]
|
| D | sparse_concat_op_gpu.cu.cc | 118 void operator()(OpKernelContext* context, const OpInputList& inds, in operator ()()
119 const OpInputList& vals, const OpInputList& shapes, in operator ()()
|
| D | identity_n_op.h | 30 OpInputList input; in Compute()
|
| D | string_join_op.cc | 38 OpInputList input_list; in Compute()
|
| D | partitioned_function_ops.cc | 124 OpInputList args; in ComputeAsync()
209 OpInputList args; in Instantiate()
|
| D | ragged_gather_op.cc | 62 OpInputList params_nested_splits_in; in Compute()
122 const Tensor& indices_in, const OpInputList& params_nested_splits_in, in MakeSplits()
|
| D | queue_op.cc | 102 OpInputList components; in ComputeAsync()
143 OpInputList components; in ComputeAsync()
|
| D | composite_tensor_ops.cc | 38 OpInputList components_in; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 47 OpInputList args;
49 OpInputList captured_inputs;
60 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions()
61 OpInputList captured_inputs, in ComputeOptions()
259 OpInputList arguments; in SetupArgs()
261 OpInputList captured_inputs; in SetupArgs()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.h | 122 const OpInputList& guaranteed_constants,
131 const OpInputList& guaranteed_constants,
142 const OpInputList& guaranteed_constants, in LookupPersistentCompilationCacheAndFillCaches()
|
| D | tpu_compile_op_common.cc | 99 OpInputList dynamic_shapes; in GetDynamicShapes()
167 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCache()
182 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCacheInternal()
236 OpInputList guaranteed_constants; in ComputeInternal()
|
| D | tpu_embedding_enqueue_ops.cc | 104 OpInputList sample_indices_or_row_splits_list; in Compute()
108 OpInputList embedding_indices_list; in Compute()
112 OpInputList aggregation_weights_list; in Compute()
|
| D | tpu_op_util.cc | 88 const OpInputList& guaranteed_constants) { in GuaranteedConstFingerprint()
107 uint64 mlir_module_fingerprint, const OpInputList& guaranteed_constants, in CreateCompilationCacheKey()
|
| D | tpu_op_util.h | 37 uint64 mlir_module_fingerprint, const OpInputList& guaranteed_constants,
|
| D | tpu_util.h | 55 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes,
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 103 const OpInputList& input_mins, const OpInputList& input_maxes, in CalculateInputAndOutputRange()
138 const OpInputList& input_mins, const OpInputList& input_maxes, in Compute()
469 OpInputList input_tensors; in Compute()
551 OpInputList input_mins, input_maxes; in Compute()
832 void CallEigenVersion(OpKernelContext* context, const OpInputList& values, in CallEigenVersion()
833 const OpInputList& input_mins, in CallEigenVersion()
834 const OpInputList& input_maxes, in CallEigenVersion()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 410 class OpInputList {
412 typedef OpArgIterator<OpInputList, const Tensor> Iterator;
413 OpInputList() : ctx_(nullptr), start_(0), stop_(0) {} in OpInputList() function
414 OpInputList(OpKernelContext* ctx, int start, int stop) in OpInputList() function
416 OpInputList& operator=(const OpInputList& other) = default;
740 Status input_list(StringPiece name, OpInputList* list);
1614 inline const Tensor& OpInputList::operator[](int i) const {
|