/external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
D | split_handler_ops.cc | 53 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()
|
D | stats_accumulator_ops.cc | 296 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/ |
D | linalg_ops.cc | 32 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()
|
D | set_ops.cc | 55 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()
|
D | array_ops.cc | 582 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, ¶ms_shape)); in __anon7c94107b1802() 1148 TF_RETURN_IF_ERROR(c->WithRankAtLeast( in __anon7c94107b1802() [all …]
|
D | math_ops.cc | 130 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()
|
D | manip_ops.cc | 34 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); in __anon323d2f510102()
|
D | encode_proto_ops.cc | 39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &input)); in __anonbf8c5fbd0102()
|
D | ragged_conversion_ops.cc | 53 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToSparseShapeFn()
|
D | ragged_array_ops.cc | 68 c->WithRankAtLeast(params_dense_values, 1, ¶ms_dense_values)); in RaggedGatherShapeFn()
|
D | resource_variable_ops.cc | 276 TF_RETURN_IF_ERROR(c->WithRankAtLeast(handle_shape_and_type[0].shape, in __anon3f9cfa090402() 280 c->WithRankAtLeast(indices_shape, batch_dims, &unused)); in __anon3f9cfa090402()
|
D | list_ops.cc | 117 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()
|
D | spectral_ops.cc | 77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out)); in RFFTShape()
|
/external/tensorflow/tensorflow/contrib/resampler/ops/ |
D | resampler_ops.cc | 32 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/ |
D | reduce_slice_ops.cc | 34 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/ |
D | unpack_path_op.cc | 38 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anona9c96fc40102() 39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, ¶ms)); in __anona9c96fc40102()
|
D | routing_function_op.cc | 57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anonb407ab430102() 58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms)); in __anonb407ab430102()
|
D | routing_gradient_op.cc | 53 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anoned6156eb0102() 54 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms)); in __anoned6156eb0102()
|
D | k_feature_routing_function_op.cc | 60 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon7e2782180102() 61 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms)); in __anon7e2782180102()
|
D | stochastic_hard_routing_gradient_op.cc | 57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); in __anon803f4efd0102() 58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms)); in __anon803f4efd0102()
|
D | hard_routing_function_op.cc | 59 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon4f866fa10102()
|
D | stochastic_hard_routing_function_op.cc | 63 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); in __anon46e5a3550102()
|
/external/tensorflow/tensorflow/contrib/coder/ops/ |
D | coder_ops.cc | 126 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); in __anon8696facd0202()
|
/external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
D | fused_conv2d_bias_activation_op.cc | 88 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &side_input_shape)); in __anon4e6e467d0102()
|
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 244 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()
|