/external/tensorflow/tensorflow/contrib/resampler/ops/ |
resampler_ops.cc | 32 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data)); 33 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &warp));
|
/external/tensorflow/tensorflow/contrib/reduce_slice_ops/ops/ |
reduce_slice_ops.cc | 34 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &handle)); 38 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &handle));
|
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
unpack_path_op.cc | 38 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); 39 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, ¶ms));
|
routing_function_op.cc | 57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); 58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms));
|
routing_gradient_op.cc | 53 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); 54 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms));
|
k_feature_routing_function_op.cc | 60 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input)); 61 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms));
|
stochastic_hard_routing_gradient_op.cc | 57 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); 58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, ¶ms));
|
hard_routing_function_op.cc | 59 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input));
|
stochastic_hard_routing_function_op.cc | 63 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input));
|
/external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
split_handler_ops.cc | 52 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); 56 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); 111 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); 115 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape)); 170 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &gradients_shape)); 174 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(4), 1, &hessians_shape));
|
stats_accumulator_ops.cc | 296 TF_RETURN_IF_ERROR(c->WithRankAtLeast( 301 TF_RETURN_IF_ERROR(c->WithRankAtLeast( 383 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(5), 2, &gradients_shape)); 387 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(6), 2, &hessians_shape));
|
/external/tensorflow/tensorflow/core/ops/ |
linalg_ops.cc | 32 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, 2, &s)); 58 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &lhs)); 60 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &rhs)); 119 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); 150 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); 195 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); 214 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input));
|
set_ops.cc | 55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); 77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input1_shape, 2, &input1_shape)); 117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape));
|
spectral_ops.cc | 71 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out));
|
math_ops.cc | 74 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &a_shape)); 75 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 2, &b_shape)); 862 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); 877 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); 901 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); 939 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &data_shape)); [all...] |
lookup_ops.cc | 162 TF_RETURN_IF_ERROR(c->WithRankAtLeast(values, 1, &values)); 180 TF_RETURN_IF_ERROR(c->WithRankAtLeast(values, 1, &values));
|
array_ops.cc | 423 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); 483 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input, split_dim + 1, &input)); 593 TF_RETURN_IF_ERROR(c->WithRankAtLeast(in, 1, &in)); 637 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &in)); 659 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input)); 660 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(1), 1, &diag)); 690 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &in)); 891 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &unused)); [all...] |
resource_variable_ops.cc | 159 c->WithRankAtLeast(handle_shape_and_type.shape, 1, &unused));
|
image_ops.cc | 99 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 1, &input));
|
/external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
fused_conv2d_bias_activation_op.cc | 88 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(3), 1, &side_input_shape));
|
/external/tensorflow/tensorflow/core/framework/ |
common_shape_fns.cc | 210 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 3, &input_shape)); 212 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); 268 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 3, &input_shape)); 271 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &input_shape)); [all...] |
common_shape_fns.h | 175 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), rank, &out));
|
shape_inference_test.cc | 388 TEST_F(ShapeInferenceTest, WithRankAtLeast) { 398 // WithRankAtLeast on a shape with unknown dimensionality always succeeds. 399 EXPECT_TRUE(c.WithRankAtLeast(in0, 1, &s1).ok()); 403 EXPECT_TRUE(c.WithRankAtLeast(in0, 2, &s2).ok()); 407 // WithRankAtLeast on shape with known dimensionality. 410 StringPiece(c.WithRankAtLeast(in1, 4, &s1).ToString()) 415 EXPECT_TRUE(c.WithRankAtLeast(in1, 3, &s1).ok()); 417 EXPECT_TRUE(c.WithRankAtLeast(in1, 2, &s1).ok()); 419 EXPECT_TRUE(c.WithRankAtLeast(in1, 0, &s1).ok()); [all...] |
shape_inference.h | 387 Status WithRankAtLeast(ShapeHandle shape, int64 rank,
|
shape_inference.cc | 354 Status InferenceContext::WithRankAtLeast(ShapeHandle shape, int64 rank, 459 TF_RETURN_IF_ERROR(WithRankAtLeast(s, rank, &s)); [all...] |