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 __anon295dc3ce0102()
57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); in __anon295dc3ce0102()
125 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); in __anon295dc3ce0202()
129 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); in __anon295dc3ce0202()
196 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); in __anon295dc3ce0302()
200 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); in __anon295dc3ce0302()
Dstats_accumulator_ops.cc296 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anone13797730702()
301 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anone13797730702()
383 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(5), 2, &gradients_shape)); in __anone13797730902()
387 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(6), 2, &hessians_shape)); in __anone13797730902()
/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 __anon4fdd0bd30202()
268 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon4fdd0bd30302()
Dset_ops.cc55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon967df2c80102()
77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input1_shape, 2, &input1_shape)); in __anon967df2c80102()
117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon967df2c80202()
Darray_ops.cc582 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anon7c94107b0a02()
643 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); in __anon7c94107b0b02()
768 TF_RETURN_IF_ERROR(c->WithRankAtLeast(in, 1, &in)); in __anon7c94107b0e02()
814 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); in __anon7c94107b1002()
836 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon7c94107b1102()
837 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &diag)); in __anon7c94107b1102()
867 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &in)); in __anon7c94107b1202()
1098 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon7c94107b1702()
1119 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &params_shape)); in __anon7c94107b1802()
1148 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anon7c94107b1802()
[all …]
Dmath_ops.cc130 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &a_shape)); in __anonb22bfa860202()
131 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &b_shape)); in __anonb22bfa860202()
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 __anonb22bfa860c02()
1463 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &b_shape)); in __anonb22bfa860c02()
1690 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anonb22bfa861402()
Dmanip_ops.cc34 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon323d2f510102()
Dencode_proto_ops.cc39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &input)); in __anonbf8c5fbd0102()
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 __anon3f9cfa090402()
280 c->WithRankAtLeast(indices_shape, batch_dims, &unused)); in __anon3f9cfa090402()
Dlist_ops.cc117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &tensor)); in __anonebbd57d80402()
320 TF_RETURN_IF_ERROR(c->WithRankAtLeast(tensor_shape, 1, &ignored)); in __anonebbd57d80902()
543 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &ignored)); in __anonebbd57d81302()
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 __anon447735090102()
33 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &warp)); in __anon447735090102()
/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 __anona9c96fc40102()
39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &params)); in __anona9c96fc40102()
Drouting_function_op.cc57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anonb407ab430102()
58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anonb407ab430102()
Drouting_gradient_op.cc53 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anoned6156eb0102()
54 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anoned6156eb0102()
Dk_feature_routing_function_op.cc60 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon7e2782180102()
61 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anon7e2782180102()
Dstochastic_hard_routing_gradient_op.cc57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon803f4efd0102()
58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &params)); in __anon803f4efd0102()
Dhard_routing_function_op.cc59 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon4f866fa10102()
Dstochastic_hard_routing_function_op.cc63 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon46e5a3550102()
/external/tensorflow/tensorflow/contrib/coder/ops/
Dcoder_ops.cc126 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); in __anon8696facd0202()
/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 __anon4e6e467d0102()
/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