Home
last modified time | relevance | path

Searched refs:OP_REQUIRES_OK (Results 1 – 25 of 609) sorted by relevance

12345678910>>...25

/external/tensorflow/tensorflow/core/kernels/
Dsummary_kernels.cc40 OP_REQUIRES_OK(ctx, ctx->input("logdir", &tmp)); in Compute()
42 OP_REQUIRES_OK(ctx, ctx->input("max_queue", &tmp)); in Compute()
44 OP_REQUIRES_OK(ctx, ctx->input("flush_millis", &tmp)); in Compute()
46 OP_REQUIRES_OK(ctx, ctx->input("filename_suffix", &tmp)); in Compute()
50 OP_REQUIRES_OK(ctx, LookupOrCreateResource<SummaryWriterInterface>( in Compute()
69 OP_REQUIRES_OK(ctx, ctx->input("db_uri", &tmp)); in Compute()
71 OP_REQUIRES_OK(ctx, ctx->input("experiment_name", &tmp)); in Compute()
73 OP_REQUIRES_OK(ctx, ctx->input("run_name", &tmp)); in Compute()
75 OP_REQUIRES_OK(ctx, ctx->input("user_name", &tmp)); in Compute()
79 OP_REQUIRES_OK( in Compute()
[all …]
Dgenerate_vocab_remapping_op.cc41 OP_REQUIRES_OK(context, in GenerateVocabRemappingOp()
43 OP_REQUIRES_OK(context, context->GetAttr("num_new_vocab", &num_new_vocab_)); in GenerateVocabRemappingOp()
44 OP_REQUIRES_OK(context, in GenerateVocabRemappingOp()
50 OP_REQUIRES_OK(context, in Compute()
69 OP_REQUIRES_OK(context, lookup::InitializeTableFromTextFile( in Compute()
83 OP_REQUIRES_OK(context, in Compute()
100 OP_REQUIRES_OK(context, in Compute()
112 OP_REQUIRES_OK( in Compute()
120 OP_REQUIRES_OK( in Compute()
127 OP_REQUIRES_OK( in Compute()
[all …]
Dcandidate_sampler_ops.cc36 OP_REQUIRES_OK(context, context->GetAttr("num_sampled", &num_sampled_)); in BaseCandidateSamplerOp()
37 OP_REQUIRES_OK(context, context->GetAttr("num_true", &num_true_)); in BaseCandidateSamplerOp()
38 OP_REQUIRES_OK(context, context->GetAttr("unique", &unique_)); in BaseCandidateSamplerOp()
39 OP_REQUIRES_OK(context, generator_.Init(context)); in BaseCandidateSamplerOp()
61 OP_REQUIRES_OK(context, in Compute()
66 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
70 OP_REQUIRES_OK(context, in Compute()
118 OP_REQUIRES_OK(context, context->GetAttr("range_max", &range_max)); in SimpleCandidateSamplerOp()
142 OP_REQUIRES_OK(context, context->GetAttr("num_sampled", &range_max)); in AllCandidateSamplerOp()
155 OP_REQUIRES_OK(context, context->GetAttr("range_max", &range_max)); in FixedUnigramCandidateSamplerOp()
[all …]
Davgpooling_op.cc60 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingOp()
68 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
72 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in AvgPoolingOp()
76 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in AvgPoolingOp()
108 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
138 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingOp()
141 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
145 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in AvgPoolingOp()
149 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in AvgPoolingOp()
180 OP_REQUIRES_OK(context, in Compute()
[all …]
Dexample_parsing_ops.cc52 OP_REQUIRES_OK(ctx, attrs_.Init(ctx, op_version_)); in ParseExampleOp()
64 OP_REQUIRES_OK(ctx, ctx->input("serialized", &serialized)); in Compute()
65 OP_REQUIRES_OK(ctx, ctx->input("names", &names)); in Compute()
67 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "dense_keys", &dense_keys_t)); in Compute()
68 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "sparse_keys", &sparse_keys_t)); in Compute()
69 OP_REQUIRES_OK(ctx, GetTensorKeys(ctx, "ragged_keys", &ragged_keys_t)); in Compute()
71 OP_REQUIRES_OK(ctx, GetInputListKeys(ctx, "dense_keys", &dense_keys_t)); in Compute()
72 OP_REQUIRES_OK(ctx, GetInputListKeys(ctx, "sparse_keys", &sparse_keys_t)); in Compute()
79 OP_REQUIRES_OK(ctx, ctx->input_list("dense_defaults", &dense_defaults)); in Compute()
82 OP_REQUIRES_OK( in Compute()
[all …]
Dctc_loss_op.cc79 OP_REQUIRES_OK(ctx, ctx->GetAttr("preprocess_collapse_repeated", in CTCLossOp()
81 OP_REQUIRES_OK(ctx, in CTCLossOp()
83 OP_REQUIRES_OK(ctx, ctx->GetAttr("ignore_longer_outputs_than_inputs", in CTCLossOp()
92 OP_REQUIRES_OK(ctx, ctx->input("inputs", &inputs)); in Compute()
93 OP_REQUIRES_OK(ctx, ctx->input("labels_indices", &labels_indices)); in Compute()
94 OP_REQUIRES_OK(ctx, ctx->input("labels_values", &labels_values)); in Compute()
95 OP_REQUIRES_OK(ctx, ctx->input("sequence_length", &seq_len)); in Compute()
142 OP_REQUIRES_OK( in Compute()
177 OP_REQUIRES_OK(ctx, ctx->allocate_output("loss", seq_len->shape(), &loss)); in Compute()
181 OP_REQUIRES_OK(ctx, in Compute()
[all …]
Dstrided_slice_op.cc85 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceOp()
86 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceOp()
87 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceOp()
88 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceOp()
89 OP_REQUIRES_OK(context, in StridedSliceOp()
102 OP_REQUIRES_OK( in Compute()
138 OP_REQUIRES_OK(context, context->allocate_output(0, final_shape, &result)); in Compute()
193 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceGradOp()
194 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceGradOp()
195 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceGradOp()
[all …]
Dreader_ops.cc34 OP_REQUIRES_OK(context, in Compute()
83 OP_REQUIRES_OK(context, in ComputeWithReader()
87 OP_REQUIRES_OK(context, in ComputeWithReader()
90 OP_REQUIRES_OK(context, in ComputeWithReader()
114 OP_REQUIRES_OK(context, context->input("num_records", &num_records_tensor)); in ComputeWithReader()
117 OP_REQUIRES_OK(context, in ComputeWithReader()
136 OP_REQUIRES_OK(context, in ComputeWithReader()
141 OP_REQUIRES_OK(context, in ComputeWithReader()
166 OP_REQUIRES_OK(context, context->allocate_output("records_produced", in ComputeWithReader()
184 OP_REQUIRES_OK(context, context->allocate_output("units_completed", in ComputeWithReader()
[all …]
Dsparse_reduce_op.cc157 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_)); in SparseReduceOp()
162 OP_REQUIRES_OK(ctx, ctx->input("input_indices", &indices_t)); in Compute()
163 OP_REQUIRES_OK(ctx, ctx->input("input_values", &values_t)); in Compute()
164 OP_REQUIRES_OK(ctx, ctx->input("input_shape", &shape_t)); in Compute()
165 OP_REQUIRES_OK(ctx, ctx->input("reduction_axes", &reduction_axes_t)); in Compute()
167 OP_REQUIRES_OK(ctx, ValidateInputs(shape_t, reduction_axes_t)); in Compute()
176 OP_REQUIRES_OK(ctx, SparseTensor::Create( in Compute()
183 OP_REQUIRES_OK( in Compute()
189 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in Compute()
253 OP_REQUIRES_OK(ctx, ctx->GetAttr("keep_dims", &keep_dims_)); in SparseReduceSparseOp()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc95 OP_REQUIRES_OK(ctx, GetLeadingDimForTensorList(ctx->Input(0), &leading_dim, in Compile()
137 OP_REQUIRES_OK(ctx, ctx->GetAttr("element_dtype", &dtype_)); in TensorListReserveOp()
147 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &num_elements)); in Compile()
149 OP_REQUIRES_OK( in Compile()
164 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type)); in Compile()
167 OP_REQUIRES_OK(ctx, in Compile()
172 OP_REQUIRES_OK(ctx, GetTensorListShapeFromElementShape( in Compile()
178 OP_REQUIRES_OK(ctx, list_dynamic_dims_or.status()); in Compile()
180 OP_REQUIRES_OK(ctx, CreateZerosTensorListWithShape( in Compile()
184 OP_REQUIRES_OK( in Compile()
[all …]
Dtraining_ops.cc37 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &handle)); in Compile()
52 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, handle)); in Compile()
76 OP_REQUIRES_OK(ctx, ctx->GetAttr("T", &dtype_)); in ResourceApplyProximalGradientDescent()
82 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, dtype_, &var_shape, &var)); in Compile()
107 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, dtype_, var)); in Compile()
120 OP_REQUIRES_OK(ctx, ctx->GetAttr("use_nesterov", &use_nesterov_)); in ResourceApplyMomentum()
128 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(0, type, &var_shape, &var)); in Compile()
129 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(1, type, &accum_shape, &accum)); in Compile()
164 OP_REQUIRES_OK(ctx, ctx->AssignVariable(0, type, var)); in Compile()
165 OP_REQUIRES_OK(ctx, ctx->AssignVariable(1, type, accum)); in Compile()
[all …]
Dtensor_array_ops.cc136 OP_REQUIRES_OK(ctx, ctx->GetAttr("element_shape", &element_shape_)); in TensorArrayOp()
137 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtype", &dtype_)); in TensorArrayOp()
139 OP_REQUIRES_OK(ctx, ctx->GetAttr("dynamic_size", &dynamic_size)); in TensorArrayOp()
145 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_array_name", &tensor_array_name_)); in TensorArrayOp()
150 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &size)); in Compile()
194 OP_REQUIRES_OK(ctx, ctx->GetAttr("T", &dtype_)); in TensorArrayWriteOp()
205 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(0, &resource)); in Compile()
206 OP_REQUIRES_OK(ctx, in Compile()
233 OP_REQUIRES_OK(ctx, resource->SetValue(written)); in Compile()
248 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtype", &dtype_)); in TensorArrayReadOp()
[all …]
Dstrided_slice_op.cc48 OP_REQUIRES_OK(ctx, ctx->GetAttr("begin_mask", &begin_mask_)); in StridedSliceOp()
49 OP_REQUIRES_OK(ctx, ctx->GetAttr("end_mask", &end_mask_)); in StridedSliceOp()
50 OP_REQUIRES_OK(ctx, ctx->GetAttr("ellipsis_mask", &ellipsis_mask_)); in StridedSliceOp()
51 OP_REQUIRES_OK(ctx, ctx->GetAttr("new_axis_mask", &new_axis_mask_)); in StridedSliceOp()
52 OP_REQUIRES_OK(ctx, ctx->GetAttr("shrink_axis_mask", &shrink_axis_mask_)); in StridedSliceOp()
53 OP_REQUIRES_OK(ctx, ctx->GetAttr("Index", &index_type_)); in StridedSliceOp()
212 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
216 OP_REQUIRES_OK( in Compile()
220 OP_REQUIRES_OK( in Compile()
223 OP_REQUIRES_OK(ctx, LiteralToHostTensor(strides_literal, index_type_, in Compile()
[all …]
Dif_op.cc31 OP_REQUIRES_OK(ctx, ctx->GetAttr("then_branch", &name_attr)); in XlaIfOp()
33 OP_REQUIRES_OK(ctx, ctx->GetAttr("else_branch", &name_attr)); in XlaIfOp()
36 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tcond", &cond_type_)); in XlaIfOp()
37 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tin", &input_types_)); in XlaIfOp()
38 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tout", &output_types_)); in XlaIfOp()
45 OP_REQUIRES_OK(ctx, ctx->GetAttr(kPropagateCompileTimeConsts, in XlaIfOp()
188 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(i + 1, &resource)); in Compile()
204 OP_REQUIRES_OK(ctx, shape_or.status()); in Compile()
216 OP_REQUIRES_OK(ctx, FindMustBeConstNodes(ctx, then_branch_, in Compile()
219 OP_REQUIRES_OK(ctx, FindMustBeConstNodes(ctx, else_branch_, in Compile()
[all …]
/external/tensorflow/tensorflow/lite/experimental/microfrontend/ops/
Daudio_microfrontend_op.cc156 OP_REQUIRES_OK(ctx, ctx->GetAttr("sample_rate", &sample_rate_)); in AudioMicrofrontendOp()
159 OP_REQUIRES_OK(ctx, ctx->GetAttr("window_size", &window_size)); in AudioMicrofrontendOp()
163 OP_REQUIRES_OK(ctx, ctx->GetAttr("window_step", &window_step)); in AudioMicrofrontendOp()
166 OP_REQUIRES_OK( in AudioMicrofrontendOp()
168 OP_REQUIRES_OK(ctx, ctx->GetAttr("upper_band_limit", in AudioMicrofrontendOp()
170 OP_REQUIRES_OK(ctx, ctx->GetAttr("lower_band_limit", in AudioMicrofrontendOp()
172 OP_REQUIRES_OK(ctx, ctx->GetAttr("smoothing_bits", in AudioMicrofrontendOp()
174 OP_REQUIRES_OK(ctx, ctx->GetAttr("even_smoothing", in AudioMicrofrontendOp()
176 OP_REQUIRES_OK(ctx, ctx->GetAttr("odd_smoothing", in AudioMicrofrontendOp()
178 OP_REQUIRES_OK(ctx, in AudioMicrofrontendOp()
[all …]
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc74 OP_REQUIRES_OK(ctx, ctx->GetAttr("cost_estimate_ns", &cost_estimate_)); in HostComputeOp()
75 OP_REQUIRES_OK(ctx, ctx->GetAttr("key", &key_)); in HostComputeOp()
76 OP_REQUIRES_OK(ctx, ctx->GetAttr("tpu_core", &tpu_core_)); in HostComputeOp()
77 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tinputs", &input_dtypes_)); in HostComputeOp()
82 OP_REQUIRES_OK(ctx, ctx->GetAttr("Toutputs", &output_dtypes_)); in HostComputeOp()
87 OP_REQUIRES_OK(ctx, ctx->GetAttr("ancestors", &ancestors_)); in HostComputeOp()
89 OP_REQUIRES_OK( in HostComputeOp()
92 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &static_output_shapes_)); in HostComputeOp()
97 OP_REQUIRES_OK(ctx, MakeXlaShapes(static_output_shapes_, output_dtypes_, in HostComputeOp()
113 OP_REQUIRES_OK(ctx, FunctionDefToBodyHelper( in HostComputeOp()
[all …]
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h189 OP_REQUIRES_OK( in Compute()
193 OP_REQUIRES_OK(ctx, DeviceAccessor::GetResourceManager(ctx, &rm)); in Compute()
198 OP_REQUIRES_OK(ctx, DeviceAccessor::InitScopedRef(ctx, &device_ref)); in Compute()
202 OP_REQUIRES_OK(ctx, XRTTupleAllocation::CreateAndTransfer( in Compute()
219 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtype", &dtype_)); in XRTAllocateUninitializedOp()
220 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape", &tf_shape_)); in XRTAllocateUninitializedOp()
221 OP_REQUIRES_OK(ctx, TensorShapeToXLAShape(dtype_, tf_shape_, &xla_shape_)); in XRTAllocateUninitializedOp()
233 OP_REQUIRES_OK(ctx, DeviceAccessor::GetResourceManager(ctx, &rm)); in Compute()
238 OP_REQUIRES_OK(ctx, DeviceAccessor::InitScopedRef(ctx, &device_ref)); in Compute()
242 OP_REQUIRES_OK(ctx, in Compute()
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresource_ops.cc44 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute()
49 OP_REQUIRES_OK(context, context->input("tree_ensemble_serialized", in Compute()
66 OP_REQUIRES_OK(context, status); in Compute()
83 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
104 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
106 OP_REQUIRES_OK(context, context->allocate_output(1, TensorShape(), in Compute()
108 OP_REQUIRES_OK(context, in Compute()
111 OP_REQUIRES_OK(context, in Compute()
114 OP_REQUIRES_OK(context, context->allocate_output( in Compute()
145 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
[all …]
Dstats_ops.cc46 OP_REQUIRES_OK(context, context->GetAttr("max_splits", &max_splits_)); in BoostedTreesCalculateBestGainsPerFeatureOp()
47 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesCalculateBestGainsPerFeatureOp()
53 OP_REQUIRES_OK(context, context->input("node_id_range", &node_id_range_t)); in Compute()
59 OP_REQUIRES_OK(context, context->input_list("stats_summary_list", in Compute()
70 OP_REQUIRES_OK(context, context->input("l1", &l1_t)); in Compute()
73 OP_REQUIRES_OK(context, context->input("l2", &l2_t)); in Compute()
76 OP_REQUIRES_OK(context, in Compute()
80 OP_REQUIRES_OK(context, in Compute()
86 OP_REQUIRES_OK( in Compute()
89 OP_REQUIRES_OK(context, in Compute()
[all …]
Dtraining_ops.cc38 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesUpdateEnsembleOp()
41 OP_REQUIRES_OK(context, context->GetAttr("pruning_mode", &pruning_index)); in BoostedTreesUpdateEnsembleOp()
48 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
60 OP_REQUIRES_OK(context, context->input_list("node_ids", &node_ids_list)); in Compute()
61 OP_REQUIRES_OK(context, context->input_list("gains", &gains_list)); in Compute()
62 OP_REQUIRES_OK(context, in Compute()
64 OP_REQUIRES_OK(context, context->input_list("left_node_contribs", in Compute()
66 OP_REQUIRES_OK(context, context->input_list("right_node_contribs", in Compute()
70 OP_REQUIRES_OK(context, context->input("feature_ids", &feature_ids_t)); in Compute()
74 OP_REQUIRES_OK(context, context->input("max_depth", &max_depth_t)); in Compute()
[all …]
Dquantile_ops.cc101 OP_REQUIRES_OK(context, context->GetAttr(kMaxElementsName, &max_elements_)); in BoostedTreesCreateQuantileStreamResourceOp()
110 OP_REQUIRES_OK(context, context->input(kEpsilonName, &epsilon_t)); in Compute()
117 OP_REQUIRES_OK(context, context->input(kNumStreamsName, &num_streams_t)); in Compute()
143 OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_)); in BoostedTreesMakeQuantileSummariesOp()
149 OP_REQUIRES_OK( in Compute()
154 OP_REQUIRES_OK(context, in Compute()
165 OP_REQUIRES_OK(context, context->input(kEpsilonName, &epsilon_t)); in Compute()
169 OP_REQUIRES_OK( in Compute()
186 OP_REQUIRES_OK( in Compute()
223 OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_)); in BoostedTreesFlushQuantileSummariesOp()
[all …]
/external/tensorflow/tensorflow/core/kernels/rnn/
Dgru_ops.cc35 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
38 OP_REQUIRES_OK(ctx, ctx->input("h_prev", &h_prev_tensor)); in Compute()
41 OP_REQUIRES_OK(ctx, ctx->input("w_ru", &w_ru_tensor)); in Compute()
44 OP_REQUIRES_OK(ctx, ctx->input("w_c", &w_c_tensor)); in Compute()
47 OP_REQUIRES_OK(ctx, ctx->input("b_ru", &b_ru_tensor)); in Compute()
50 OP_REQUIRES_OK(ctx, ctx->input("b_c", &b_c_tensor)); in Compute()
110 OP_REQUIRES_OK( in Compute()
115 OP_REQUIRES_OK( in Compute()
120 OP_REQUIRES_OK( in Compute()
125 OP_REQUIRES_OK(ctx, ctx->forward_input_or_allocate_output( in Compute()
[all …]
Dlstm_ops.cc301 OP_REQUIRES_OK(ctx, ctx->GetAttr("forget_bias", &forget_bias_)); in LSTMBlockCellOp()
302 OP_REQUIRES_OK(ctx, ctx->GetAttr("cell_clip", &cell_clip_)); in LSTMBlockCellOp()
303 OP_REQUIRES_OK(ctx, ctx->GetAttr("use_peephole", &use_peephole_)); in LSTMBlockCellOp()
308 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
311 OP_REQUIRES_OK(ctx, ctx->input("cs_prev", &cs_prev_tensor)); in Compute()
314 OP_REQUIRES_OK(ctx, ctx->input("h_prev", &h_prev_tensor)); in Compute()
317 OP_REQUIRES_OK(ctx, ctx->input("w", &w_tensor)); in Compute()
320 OP_REQUIRES_OK(ctx, ctx->input("wci", &wci_tensor)); in Compute()
323 OP_REQUIRES_OK(ctx, ctx->input("wcf", &wcf_tensor)); in Compute()
326 OP_REQUIRES_OK(ctx, ctx->input("wco", &wco_tensor)); in Compute()
[all …]
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_configuration_ops.cc116 OP_REQUIRES_OK(ctx, num_devices_per_host.status()); in Compute()
122 OP_REQUIRES_OK(ctx, CreateTpuCompilationCache(rmgr, &compilation_cache)); in Compute()
126 OP_REQUIRES_OK( in Compute()
130 OP_REQUIRES_OK(ctx, DeleteIfExists<tpu::TpuMeshStateInterface>( in Compute()
134 OP_REQUIRES_OK( in Compute()
139 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &ctx_output)); in Compute()
192 OP_REQUIRES_OK(ctx, GetTpuMeshStateInterface(rmgr, &mesh_state)); in Compute()
198 OP_REQUIRES_OK(ctx, GetTPUPodState(rmgr, &pod_state)); in Compute()
225 OP_REQUIRES_OK(ctx, StatusFromTF_Status(status)); in Compute()
228 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &ctx_output)); in Compute()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_ops.cc56 OP_REQUIRES_OK(ctx, in DatasetToGraphOp()
64 OP_REQUIRES_OK(ctx, ctx->GetAttr(kAllowStateful, &allow_stateful)); in DatasetToGraphOp()
76 OP_REQUIRES_OK( in DatasetToGraphOp()
83 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
108 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &result)); in Compute()
114 OP_REQUIRES_OK(ctx, GetDatasetFromVariantTensor(ctx->input(0), &dataset)); in Compute()
116 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, TensorShape({}), &result)); in Compute()
122 OP_REQUIRES_OK(ctx, in Compute()
134 OP_REQUIRES_OK(ctx, ImportGraphDef({}, graph_def, &graph, nullptr)); in Compute()
139 OP_REQUIRES_OK(ctx, in Compute()
[all …]

12345678910>>...25