HomeSort by relevance Sort by last modified time
    Searched refs:WithRankAtLeast (Results 1 - 25 of 27) sorted by null

1 2

  /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, &params));
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, &params));
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, &params));
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, &params));
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, &params));
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...]

Completed in 666 milliseconds

1 2