Home
last modified time | relevance | path

Searched refs:WithRankAtLeast (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dsplit_handler_ops.cc53 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); in __anond11615e50102()
57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); in __anond11615e50102()
125 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); in __anond11615e50202()
129 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); in __anond11615e50202()
196 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); in __anond11615e50302()
200 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); in __anond11615e50302()
Dstats_accumulator_ops.cc296 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anon72691d0a0702()
301 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anon72691d0a0702()
383 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(5), 2, &gradients_shape)); in __anon72691d0a0902()
387 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(6), 2, &hessians_shape)); in __anon72691d0a0902()
/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops.cc32 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &s)); in MakeBatchSquareMatrix()
58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &lhs)); in MatrixSolveShapeFn()
60 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &rhs)); in MatrixSolveShapeFn()
117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in LuShapeFn()
143 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in QrShapeFn()
174 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in SvdShapeFn()
217 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &lhs)); in TridiagonalSolveShapeFn()
218 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &rhs)); in TridiagonalSolveShapeFn()
249 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon8ebee8ea0202()
268 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon8ebee8ea0302()
Dset_ops.cc55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon9b840f3f0102()
77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input1_shape, 2, &input1_shape)); in __anon9b840f3f0102()
117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon9b840f3f0202()
Darray_ops.cc582 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anondb9326b20a02()
643 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anondb9326b20b02()
768 TF_RETURN_IF_ERROR(c->WithRankAtLeast(in, 1, &in)); in __anondb9326b20e02()
814 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); in __anondb9326b21002()
836 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anondb9326b21102()
837 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &diag)); in __anondb9326b21102()
867 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &in)); in __anondb9326b21202()
1098 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anondb9326b21702()
1119 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &params_shape)); in __anondb9326b21802()
1148 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anondb9326b21802()
[all …]
Dmath_ops.cc130 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &a_shape)); in __anon57f5a5dd0202()
131 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &b_shape)); in __anon57f5a5dd0202()
988 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SegmentReductionShapeFn()
1003 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionShapeFn()
1027 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionGradShapeFn()
1065 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1462 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &a_shape)); in __anon57f5a5dd0c02()
1463 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &b_shape)); in __anon57f5a5dd0c02()
1690 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon57f5a5dd1402()
Dmanip_ops.cc34 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon913c45880102()
Dencode_proto_ops.cc39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &input)); in __anond41b72140102()
Dragged_conversion_ops.cc53 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToSparseShapeFn()
Dragged_array_ops.cc68 c->WithRankAtLeast(params_dense_values, 1, &params_dense_values)); in RaggedGatherShapeFn()
Dresource_variable_ops.cc276 TF_RETURN_IF_ERROR(c->WithRankAtLeast(handle_shape_and_type[0].shape, in __anon4d581ac00402()
280 c->WithRankAtLeast(indices_shape, batch_dims, &unused)); in __anon4d581ac00402()
Dlist_ops.cc117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &tensor)); in __anon9187032f0402()
320 TF_RETURN_IF_ERROR(c->WithRankAtLeast(tensor_shape, 1, &ignored)); in __anon9187032f0902()
543 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &ignored)); in __anon9187032f1302()
Dspectral_ops.cc77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out)); in RFFTShape()
/external/tensorflow/tensorflow/contrib/resampler/ops/
Dresampler_ops.cc32 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data)); in __anon6f0660200102()
33 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &warp)); in __anon6f0660200102()
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/
Dreduce_slice_ops.cc34 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &handle)); in ReduceSliceShapeFn()
38 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &handle)); in ReduceSliceShapeFn()
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
Dunpack_path_op.cc38 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon3ff10bbb0102()
39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &params)); in __anon3ff10bbb0102()
Drouting_gradient_op.cc53 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon43ac05420102()
54 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anon43ac05420102()
Drouting_function_op.cc57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon0a52599a0102()
58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anon0a52599a0102()
Dk_feature_routing_function_op.cc60 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon0ea5492f0102()
61 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anon0ea5492f0102()
Dstochastic_hard_routing_gradient_op.cc57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon2a31cb540102()
58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anon2a31cb540102()
Dhard_routing_function_op.cc59 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anond09eac580102()
Dstochastic_hard_routing_function_op.cc63 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anonf0d81fac0102()
/external/tensorflow/tensorflow/contrib/coder/ops/
Dcoder_ops.cc126 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); in __anonfb33fee40202()
/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc88 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &side_input_shape)); in __anon0cecbc340102()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc244 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 3, &input_shape)); in BiasAddShape()
246 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); in BiasAddShape()
302 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 3, &input_shape)); in BiasAddGradShape()
305 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); in BiasAddGradShape()
1259 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, min_rank, &input)); in ConcatShapeHelper()
1272 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, min_rank, &input)); in ConcatShapeHelper()
1545 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &indices_shape)); in ScatterNdUpdateShape()
1547 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(2), 1, &updates_shape)); in ScatterNdUpdateShape()

12