Home
last modified time | relevance | path

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

12345678910>>...23

/external/tensorflow/tensorflow/core/kernels/
Dsummary_kernels.cc38 OP_REQUIRES_OK(ctx, ctx->input("logdir", &tmp)); in Compute()
40 OP_REQUIRES_OK(ctx, ctx->input("max_queue", &tmp)); in Compute()
42 OP_REQUIRES_OK(ctx, ctx->input("flush_millis", &tmp)); in Compute()
44 OP_REQUIRES_OK(ctx, ctx->input("filename_suffix", &tmp)); in Compute()
48 OP_REQUIRES_OK(ctx, LookupOrCreateResource<SummaryWriterInterface>( in Compute()
68 OP_REQUIRES_OK(ctx, ctx->input("db_uri", &tmp)); in Compute()
70 OP_REQUIRES_OK(ctx, ctx->input("experiment_name", &tmp)); in Compute()
72 OP_REQUIRES_OK(ctx, ctx->input("run_name", &tmp)); in Compute()
74 OP_REQUIRES_OK(ctx, ctx->input("user_name", &tmp)); in Compute()
78 OP_REQUIRES_OK( in Compute()
[all …]
Dexample_parsing_ops.cc40 OP_REQUIRES_OK(ctx, attrs_.Init(ctx)); in ParseExampleOp()
51 OP_REQUIRES_OK(ctx, ctx->input("names", &names)); in Compute()
52 OP_REQUIRES_OK(ctx, ctx->input("serialized", &serialized)); in Compute()
53 OP_REQUIRES_OK(ctx, ctx->input_list("dense_keys", &dense_keys)); in Compute()
54 OP_REQUIRES_OK(ctx, ctx->input_list("sparse_keys", &sparse_keys)); in Compute()
55 OP_REQUIRES_OK(ctx, ctx->input_list("dense_defaults", &dense_defaults)); in Compute()
140 OP_REQUIRES_OK( in Compute()
150 OP_REQUIRES_OK(ctx, ctx->output_list("dense_values", &dense_values)); in Compute()
151 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_indices", &sparse_indices)); in Compute()
152 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_values", &sparse_values)); 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()
82 OP_REQUIRES_OK(context, in Compute()
99 OP_REQUIRES_OK(context, in Compute()
110 OP_REQUIRES_OK( in Compute()
118 OP_REQUIRES_OK( in Compute()
125 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 …]
Dquantize_and_dequantize_op.cc46 OP_REQUIRES_OK(ctx, ctx->GetAttr("signed_input", &signed_input_)); in QuantizeAndDequantizeV2Op()
47 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_bits", &num_bits_)); in QuantizeAndDequantizeV2Op()
51 OP_REQUIRES_OK(ctx, ctx->GetAttr("range_given", &range_given_)); in QuantizeAndDequantizeV2Op()
54 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in QuantizeAndDequantizeV2Op()
73 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, input.shape(), &output)); in Compute()
86 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in Compute()
88 OP_REQUIRES_OK(ctx, ctx->allocate_temp(DataTypeToEnum<T>::value, in Compute()
117 OP_REQUIRES_OK(ctx, ctx->GetAttr("signed_input", &signed_input_)); in QuantizeAndDequantizeV3Op()
118 OP_REQUIRES_OK(ctx, ctx->GetAttr("range_given", &range_given_)); in QuantizeAndDequantizeV3Op()
125 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, input.shape(), &output)); in Compute()
[all …]
Dserialize_sparse_op.cc59 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
60 OP_REQUIRES_OK(context, context->input("sparse_values", &input_values)); in Compute()
61 OP_REQUIRES_OK(context, context->input("sparse_shape", &input_shape)); in Compute()
78 OP_REQUIRES_OK(context, Initialize(&serialized_sparse)); in Compute()
81 OP_REQUIRES_OK(context, Serialize(*input_indices, &serialized_sparse_t(0))); in Compute()
82 OP_REQUIRES_OK(context, Serialize(*input_values, &serialized_sparse_t(1))); in Compute()
83 OP_REQUIRES_OK(context, Serialize(*input_shape, &serialized_sparse_t(2))); in Compute()
165 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
166 OP_REQUIRES_OK(context, context->input("sparse_values", &input_values)); in Compute()
167 OP_REQUIRES_OK(context, context->input("sparse_shape", &input_shape)); in Compute()
[all …]
Dsparse_fill_empty_rows_op.cc47 OP_REQUIRES_OK(context, context->input("indices", &indices_t)); in Compute()
48 OP_REQUIRES_OK(context, context->input("values", &values_t)); in Compute()
49 OP_REQUIRES_OK(context, context->input("dense_shape", &dense_shape_t)); in Compute()
50 OP_REQUIRES_OK(context, context->input("default_value", &default_value_t)); in Compute()
79 OP_REQUIRES_OK(context, context->allocate_output("empty_row_indicator", in Compute()
84 OP_REQUIRES_OK( in Compute()
99 OP_REQUIRES_OK(context, context->allocate_output("output_indices", in Compute()
103 OP_REQUIRES_OK(context, in Compute()
112 OP_REQUIRES_OK(context, in Compute()
142 OP_REQUIRES_OK(context, context->allocate_output("output_indices", in Compute()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc37 OP_REQUIRES_OK(ctx, ctx->GetAttr("begin_mask", &begin_mask_)); in StridedSliceOp()
38 OP_REQUIRES_OK(ctx, ctx->GetAttr("end_mask", &end_mask_)); in StridedSliceOp()
39 OP_REQUIRES_OK(ctx, ctx->GetAttr("ellipsis_mask", &ellipsis_mask_)); in StridedSliceOp()
40 OP_REQUIRES_OK(ctx, ctx->GetAttr("new_axis_mask", &new_axis_mask_)); in StridedSliceOp()
41 OP_REQUIRES_OK(ctx, ctx->GetAttr("shrink_axis_mask", &shrink_axis_mask_)); in StridedSliceOp()
42 OP_REQUIRES_OK(ctx, ctx->GetAttr("Index", &index_type_)); in StridedSliceOp()
54 OP_REQUIRES_OK(ctx, ctx->ConstantInput(1, &begin_literal)); in Compile()
55 OP_REQUIRES_OK(ctx, ctx->ConstantInput(2, &end_literal)); in Compile()
56 OP_REQUIRES_OK(ctx, ctx->ConstantInput(3, &strides_literal)); in Compile()
59 OP_REQUIRES_OK( in Compile()
[all …]
Dtensor_list_ops.cc51 OP_REQUIRES_OK(ctx, GetTensorListPushIndex(ctx->Input(0), &index)); in Compile()
101 OP_REQUIRES_OK(ctx, ctx->GetAttr("element_dtype", &dtype_)); in TensorListReserveOp()
106 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &num_elements)); in Compile()
109 OP_REQUIRES_OK(ctx, CreateZerosList(ctx, 0, num_elements, dtype_, &buffer)); in Compile()
112 OP_REQUIRES_OK( in Compile()
133 OP_REQUIRES_OK(ctx, ctx->GetAttr("element_dtype", &dtype_)); in EmptyTensorListOp()
138 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &max_num_elements)); in Compile()
145 OP_REQUIRES_OK(ctx, in Compile()
149 OP_REQUIRES_OK( in Compile()
170 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape_type", &shape_type_)); in TensorListElementShapeOp()
[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 …]
Dif_op.cc29 OP_REQUIRES_OK(ctx, ctx->GetAttr("then_branch", &name_attr)); in XlaIfOp()
31 OP_REQUIRES_OK(ctx, ctx->GetAttr("else_branch", &name_attr)); in XlaIfOp()
34 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tcond", &cond_type_)); in XlaIfOp()
35 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tin", &input_types_)); in XlaIfOp()
36 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tout", &output_types_)); in XlaIfOp()
66 OP_REQUIRES_OK(ctx, ctx->GetResourceInput(i + 1, &resource)); in Compile()
106 OP_REQUIRES_OK(ctx, compiler->CompileFunction(options, then_branch_, in Compile()
109 OP_REQUIRES_OK(ctx, compiler->CompileFunction(options, else_branch_, in Compile()
116 OP_REQUIRES_OK(ctx, in Compile()
126 OP_REQUIRES_OK(ctx, resource->GetOrCreateTensorArrayGradient( in Compile()
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc29 OP_REQUIRES_OK(context, context->GetAttr("max_splits", &max_splits_)); in BoostedTreesCalculateBestGainsPerFeatureOp()
30 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesCalculateBestGainsPerFeatureOp()
36 OP_REQUIRES_OK(context, context->input("node_id_range", &node_id_range_t)); in Compute()
42 OP_REQUIRES_OK(context, context->input_list("stats_summary_list", in Compute()
51 OP_REQUIRES_OK(context, context->input("l1", &l1_t)); in Compute()
54 OP_REQUIRES_OK(context, context->input("l2", &l2_t)); in Compute()
57 OP_REQUIRES_OK(context, in Compute()
61 OP_REQUIRES_OK(context, in Compute()
67 OP_REQUIRES_OK( in Compute()
70 OP_REQUIRES_OK(context, in Compute()
[all …]
Dresource_ops.cc43 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute()
48 OP_REQUIRES_OK(context, context->input("tree_ensemble_serialized", in Compute()
65 OP_REQUIRES_OK(context, status); in Compute()
82 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 …]
Dquantile_ops.cc100 OP_REQUIRES_OK(context, context->GetAttr(kMaxElementsName, &max_elements_)); in BoostedTreesCreateQuantileStreamResourceOp()
109 OP_REQUIRES_OK(context, context->input(kEpsilonName, &epsilon_t)); in Compute()
116 OP_REQUIRES_OK(context, context->input(kNumStreamsName, &num_streams_t)); in Compute()
142 OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_)); in BoostedTreesMakeQuantileSummariesOp()
148 OP_REQUIRES_OK( in Compute()
153 OP_REQUIRES_OK(context, in Compute()
164 OP_REQUIRES_OK(context, context->input(kEpsilonName, &epsilon_t)); in Compute()
168 OP_REQUIRES_OK( in Compute()
185 OP_REQUIRES_OK( in Compute()
225 OP_REQUIRES_OK(context, in Compute()
[all …]
Dtraining_ops.cc48 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesUpdateEnsembleOp()
51 OP_REQUIRES_OK(context, context->GetAttr("pruning_mode", &pruning_index)); in BoostedTreesUpdateEnsembleOp()
58 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
71 OP_REQUIRES_OK(context, context->input_list("node_ids", &node_ids_list)); in Compute()
72 OP_REQUIRES_OK(context, context->input_list("gains", &gains_list)); in Compute()
73 OP_REQUIRES_OK(context, in Compute()
75 OP_REQUIRES_OK(context, context->input_list("left_node_contribs", in Compute()
77 OP_REQUIRES_OK(context, context->input_list("right_node_contribs", in Compute()
81 OP_REQUIRES_OK(context, context->input("feature_ids", &feature_ids_t)); in Compute()
85 OP_REQUIRES_OK(context, context->input("max_depth", &max_depth_t)); in Compute()
[all …]
Dprediction_ops.cc47 OP_REQUIRES_OK(context, context->GetAttr("num_bucketized_features", in BoostedTreesTrainingPredictOp()
49 OP_REQUIRES_OK(context, in BoostedTreesTrainingPredictOp()
59 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
66 OP_REQUIRES_OK(context, context->input_list("bucketized_features", in Compute()
76 OP_REQUIRES_OK(context, in Compute()
81 OP_REQUIRES_OK(context, in Compute()
87 OP_REQUIRES_OK(context, in Compute()
94 OP_REQUIRES_OK(context, context->allocate_output("tree_ids", {batch_size}, in Compute()
99 OP_REQUIRES_OK(context, context->allocate_output("node_ids", {batch_size}, in Compute()
189 OP_REQUIRES_OK(context, context->GetAttr("num_bucketized_features", in BoostedTreesPredictOp()
[all …]
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h184 OP_REQUIRES_OK( in Compute()
188 OP_REQUIRES_OK(ctx, DeviceAccessor::GetResourceManager(ctx, &rm)); in Compute()
193 OP_REQUIRES_OK(ctx, DeviceAccessor::InitScopedRef(ctx, &device_ref)); in Compute()
196 OP_REQUIRES_OK(ctx, XRTTupleAllocation::CreateAndTransfer( in Compute()
202 OP_REQUIRES_OK(ctx, allocation->Intern(rm, &key)); in Compute()
217 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &tf_shapes_)); in XRTAllocateFromTensorOp()
218 OP_REQUIRES_OK(ctx, ctx->GetAttr("dtypes", &dtypes_)); in XRTAllocateFromTensorOp()
219 OP_REQUIRES_OK(ctx, ctx->GetAttr("make_tuple", &make_tuple)); in XRTAllocateFromTensorOp()
222 OP_REQUIRES_OK(ctx, ctx->GetAttr("layouts", &minor_to_major)); in XRTAllocateFromTensorOp()
231 OP_REQUIRES_OK( in XRTAllocateFromTensorOp()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc137 OP_REQUIRES_OK(context, context->allocate_output("output_partition_ids", in SerializeScalarAccumulatorToOutput()
144 OP_REQUIRES_OK(context, context->allocate_output("output_feature_ids", in SerializeScalarAccumulatorToOutput()
150 OP_REQUIRES_OK( in SerializeScalarAccumulatorToOutput()
156 OP_REQUIRES_OK( in SerializeScalarAccumulatorToOutput()
178 OP_REQUIRES_OK(context, context->allocate_output("output_partition_ids", in SerializeTensorAccumulatorToOutput()
184 OP_REQUIRES_OK(context, context->allocate_output("output_feature_ids", in SerializeTensorAccumulatorToOutput()
193 OP_REQUIRES_OK(context, in SerializeTensorAccumulatorToOutput()
202 OP_REQUIRES_OK(context, context->allocate_output("output_hessians", in SerializeTensorAccumulatorToOutput()
254 OP_REQUIRES_OK(context, context->input("partition_ids", &partition_ids_t)); in AddToScalarAccumulator()
256 OP_REQUIRES_OK(context, context->input("feature_ids", &feature_ids_t)); in AddToScalarAccumulator()
[all …]
Dquantile_ops.cc108 OP_REQUIRES_OK(context, context->output_list(output_name, &output_list)); in QuantizeFeatures()
117 OP_REQUIRES_OK( in QuantizeFeatures()
164 OP_REQUIRES_OK(context, context->GetAttr(name, &serialized_config)); in ParseConfig()
201 OP_REQUIRES_OK( in CopyBoundaries()
233 OP_REQUIRES_OK(context, context->GetAttr(kEpsilonName, &epsilon_)); in CreateQuantileAccumulatorOp()
234 OP_REQUIRES_OK(context, in CreateQuantileAccumulatorOp()
236 OP_REQUIRES_OK(context, context->GetAttr(kMaxElementsName, &max_elements_)); in CreateQuantileAccumulatorOp()
237 OP_REQUIRES_OK(context, in CreateQuantileAccumulatorOp()
245 OP_REQUIRES_OK(context, context->input(kStampTokenName, &stamp_token_t)); in Compute()
281 OP_REQUIRES_OK(context, context->input_list(kResourceHandlesName, in Compute()
[all …]
Dprediction_ops.cc96 OP_REQUIRES_OK(context, context->GetAttr("use_locking", &use_locking_)); in GradientTreesPredictionOp()
98 OP_REQUIRES_OK(context, context->GetAttr("center_bias", &center_bias_)); in GradientTreesPredictionOp()
100 OP_REQUIRES_OK( in GradientTreesPredictionOp()
105 OP_REQUIRES_OK(context, context->GetAttr(kLearnerConfigAttributeName, in GradientTreesPredictionOp()
119 OP_REQUIRES_OK(context, context->GetAttr("reduce_dim", &reduce_dim)); in GradientTreesPredictionOp()
133 OP_REQUIRES_OK(context, context->GetAttr(kApplyAveragingAttributeName, in GradientTreesPredictionOp()
168 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
192 OP_REQUIRES_OK(context, TensorUtils::ReadDenseFloatFeatures( in DoCompute()
199 OP_REQUIRES_OK(context, TensorUtils::ReadSparseFloatFeatures( in DoCompute()
208 OP_REQUIRES_OK(context, TensorUtils::ReadSparseIntFeatures( in DoCompute()
[all …]
Dmodel_ops.cc40 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute()
45 OP_REQUIRES_OK(context, context->input("tree_ensemble_config", in Compute()
73 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
78 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
92 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
97 OP_REQUIRES_OK(context, context->allocate_output(0, TensorShape(), in Compute()
101 OP_REQUIRES_OK( in Compute()
116 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
123 OP_REQUIRES_OK(context, context->input("stamp_token", &stamp_token_t)); in Compute()
128 OP_REQUIRES_OK(context, context->input("tree_ensemble_config", in Compute()
[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/contrib/rnn/kernels/
Dlstm_ops.cc238 OP_REQUIRES_OK(ctx, ctx->GetAttr("forget_bias", &forget_bias_)); in LSTMBlockCellOp()
239 OP_REQUIRES_OK(ctx, ctx->GetAttr("cell_clip", &cell_clip_)); in LSTMBlockCellOp()
240 OP_REQUIRES_OK(ctx, ctx->GetAttr("use_peephole", &use_peephole_)); in LSTMBlockCellOp()
245 OP_REQUIRES_OK(ctx, ctx->input("x", &x_tensor)); in Compute()
248 OP_REQUIRES_OK(ctx, ctx->input("cs_prev", &cs_prev_tensor)); in Compute()
251 OP_REQUIRES_OK(ctx, ctx->input("h_prev", &h_prev_tensor)); in Compute()
254 OP_REQUIRES_OK(ctx, ctx->input("w", &w_tensor)); in Compute()
257 OP_REQUIRES_OK(ctx, ctx->input("wci", &wci_tensor)); in Compute()
260 OP_REQUIRES_OK(ctx, ctx->input("wcf", &wcf_tensor)); in Compute()
263 OP_REQUIRES_OK(ctx, ctx->input("wco", &wco_tensor)); in Compute()
[all …]
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 …]

12345678910>>...23