| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| fill_op.cc | 40 ctx, TensorShapeUtils::IsVector(dims_shape),
43 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(value_shape),
|
| one_hot_op.cc | 46 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(depth_shape),
49 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(on_value_shape),
52 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(off_value_shape),
|
| select_op.cc | 49 bool cond_is_scalar = TensorShapeUtils::IsScalar(cond_shape);
51 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(cond_shape),
55 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then_shape),
|
| training_ops.cc | 40 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape),
85 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape),
89 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape),
93 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape),
137 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
148 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(momentum_shape),
195 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
206 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(momentum_shape),
251 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
293 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape)
[all...] |
| clip_by_value_op.cc | 47 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(min_shape), shape_error());
51 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(max_shape), shape_error());
|
| bias_ops.cc | 45 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(input_shape),
48 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(bias_shape),
93 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(out_backprop_shape),
|
| dynamic_slice_ops.cc | 46 TensorShapeUtils::IsVector(index_shape) &&
80 TensorShapeUtils::IsVector(start_indices_shape) &&
88 TensorShapeUtils::IsVector(size_indices_shape) &&
|
| matrix_band_part_op.cc | 34 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape),
40 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_lower_in_shape),
45 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_upper_in_shape),
|
| empty_op.cc | 45 ctx, TensorShapeUtils::IsVector(shape_shape),
|
| pad_op.cc | 38 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2,
74 ctx, TensorShapeUtils::IsScalar(ctx->InputShape("constant_values")),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| string_join_op.cc | 46 is_scalar.push_back(TensorShapeUtils::IsScalar(input.shape()));
47 if (!TensorShapeUtils::IsScalar(input.shape())) {
48 if (TensorShapeUtils::IsScalar(input_shape)) {
|
| sparse_slice_op.cc | 37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()),
41 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values.shape()),
45 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_shape.shape()),
49 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_start.shape()),
53 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_size.shape()),
|
| sparse_reorder_op.cc | 40 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_ind.shape()),
46 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_val.shape()),
52 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_shape_in.shape()),
|
| matching_files_op.cc | 38 TensorShapeUtils::IsScalar(patterns_t->shape()) ||
39 TensorShapeUtils::IsVector(patterns_t->shape()),
|
| sparse_add_op.cc | 38 TensorShapeUtils::IsMatrix(a_indices->shape()) &&
39 TensorShapeUtils::IsMatrix(b_indices->shape()),
51 TensorShapeUtils::IsVector(a_values_t->shape()) &&
52 TensorShapeUtils::IsVector(b_values_t->shape()),
68 TensorShapeUtils::IsVector(a_shape->shape()) &&
69 TensorShapeUtils::IsVector(b_shape->shape()),
89 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(thresh_t->shape()),
|
| sparse_split_op.cc | 38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()),
42 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values.shape()),
46 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_shape.shape()),
|
| sparse_add_grad_op.cc | 39 TensorShapeUtils::IsMatrix(a_indices->shape()) &&
40 TensorShapeUtils::IsMatrix(b_indices->shape()) &&
41 TensorShapeUtils::IsMatrix(sum_indices->shape()),
48 ctx, TensorShapeUtils::IsVector(backprop_val_grad->shape()),
|
| sparse_slice_grad_op.cc | 37 TensorShapeUtils::IsMatrix(input_indices->shape()) &&
38 TensorShapeUtils::IsMatrix(output_indices->shape()),
45 ctx, TensorShapeUtils::IsVector(backprop_val_grad->shape()),
67 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(input_start->shape()),
|
| tensor_flag_utils.cc | 25 if (TensorShapeUtils::IsScalar(config.shape())) {
35 if (!TensorShapeUtils::IsMatrix(config.shape())) {
89 if (TensorShapeUtils::IsScalar(config.shape())) {
99 if (!TensorShapeUtils::IsMatrix(config.shape())) {
|
| unravel_index_op.cc | 43 TensorShapeUtils::IsVector(indices_tensor.shape()) ||
44 TensorShapeUtils::IsScalar(indices_tensor.shape()),
51 ctx, TensorShapeUtils::IsVector(dims_tensor.shape()),
82 if (TensorShapeUtils::IsScalar(indices_tensor.shape())) {
|
| edit_distance_op.cc | 43 if (!TensorShapeUtils::IsMatrix(hypothesis_indices.shape()))
47 if (!TensorShapeUtils::IsMatrix(truth_indices.shape()))
51 if (!TensorShapeUtils::IsVector(hypothesis_values.shape()))
55 if (!TensorShapeUtils::IsVector(truth_values.shape()))
59 if (!TensorShapeUtils::IsVector(hypothesis_shape.shape()))
63 if (!TensorShapeUtils::IsVector(truth_shape.shape()))
124 ctx, TensorShapeUtils::MakeShape(hypothesis_shape->vec<int64>().data(),
128 OP_REQUIRES_OK(ctx, TensorShapeUtils::MakeShape(
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| batch_features.cc | 48 TensorShapeUtils::IsMatrix(dense_float_feature.shape()),
78 TensorShapeUtils::IsMatrix(sparse_float_feature_indices.shape()),
82 TensorShapeUtils::IsVector(sparse_float_feature_values.shape()),
86 TensorShapeUtils::IsVector(sparse_float_feature_shape.shape()),
122 TensorShapeUtils::IsMatrix(sparse_int_feature_indices.shape()),
126 TensorShapeUtils::IsVector(sparse_int_feature_values.shape()),
129 TensorShapeUtils::IsVector(sparse_int_feature_shape.shape()),
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| encode_audio_op.cc | 73 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(contents.shape()),
82 OP_REQUIRES(context, TensorShapeUtils::IsScalar(file_format_tensor.shape()),
87 TensorShapeUtils::IsScalar(samples_per_second_tensor.shape()),
92 TensorShapeUtils::IsScalar(bits_per_second_tensor.shape()),
178 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(contents.shape()),
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| wals_solver_ops.cc | 45 using tensorflow::TensorShapeUtils;
83 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(factors.shape()),
85 OP_REQUIRES(context, TensorShapeUtils::IsVector(factor_weights.shape()),
88 context, TensorShapeUtils::IsScalar(unobserved_weights.shape()),
90 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_weights.shape()),
92 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()),
97 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values.shape()),
99 OP_REQUIRES(context, TensorShapeUtils::IsVector(entry_weights.shape()),
104 OP_REQUIRES(context, TensorShapeUtils::IsScalar(input_block_size.shape()),
107 context, TensorShapeUtils::IsScalar(input_is_transpose.shape())
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| tensor_shape_test.cc | 604 TensorShapeUtils::MakeShape(overflow, &shape).code());
626 EXPECT_TRUE(TensorShapeUtils::StartsWith(TensorShape({}), TensorShape({})));
628 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({})));
630 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({2})));
632 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({2, 3})));
633 EXPECT_TRUE(TensorShapeUtils::StartsWith(TensorShape({2, 3, 4}),
636 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({3})));
638 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({2, 4})));
639 EXPECT_FALSE(TensorShapeUtils::StartsWith(TensorShape({2, 3}),
641 EXPECT_FALSE(TensorShapeUtils::StartsWith(TensorShape({2, 3, 4})
[all...] |
