Searched refs:TensorShapeUtils (Results 1 - 25 of 197) sorted by relevance

12345678

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Done_hot_op.cc46 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),
H A Dselect_op.cc50 bool cond_is_scalar = TensorShapeUtils::IsScalar(cond_shape);
52 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(cond_shape),
56 OP_REQUIRES(ctx, TensorShapeUtils::IsVectorOrHigher(then_shape),
H A Dtraining_ops.cc40 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(alpha_shape),
81 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
92 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(momentum_shape),
139 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
184 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta1_power_shape),
187 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta2_power_shape),
190 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(lr_shape),
193 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta1_shape),
196 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(beta2_shape),
199 OP_REQUIRES(ctx, TensorShapeUtils
[all...]
H A Dbias_ops.cc44 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(input_shape),
47 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(bias_shape),
90 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrixOrHigher(out_backprop_shape),
H A Dmatrix_band_part_op.cc32 OP_REQUIRES(context, TensorShapeUtils::IsMatrixOrHigher(input_shape),
38 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_lower_in_shape),
43 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_upper_in_shape),
H A Dpad_op.cc36 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2,
75 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(ctx->InputShape(2)),
/external/tensorflow/tensorflow/core/kernels/
H A Dstring_join_op.cc46 is_scalar.push_back(TensorShapeUtils::IsScalar(input.shape()));
47 if (!TensorShapeUtils::IsScalar(input.shape())) {
48 if (TensorShapeUtils::IsScalar(input_shape)) {
H A Dsparse_slice_op.cc37 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()),
H A Dsparse_reorder_op.cc40 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()),
H A Dmatching_files_op.cc38 TensorShapeUtils::IsScalar(patterns_t->shape()) ||
39 TensorShapeUtils::IsVector(patterns_t->shape()),
H A Dsparse_add_op.cc38 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()),
H A Dsparse_split_op.cc38 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()),
H A Dsparse_add_grad_op.cc39 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()),
H A Dunravel_index_op.cc43 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())) {
H A Dedit_distance_op.cc43 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(
H A Dlinalg_ops_common.cc38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[0]),
49 OP_REQUIRES(context, TensorShapeUtils::IsSquareMatrix(input_matrix_shapes[0]),
60 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[0]),
62 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[1]),
78 context, TensorShapeUtils::IsSquareMatrix(input_matrix_shapes[0]),
80 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[1]),
H A Dsparse_sparse_binary_op_shared.cc136 TensorShapeUtils::IsMatrix(a_indices_t->shape()) &&
137 TensorShapeUtils::IsMatrix(b_indices_t->shape()),
143 TensorShapeUtils::IsVector(a_values_t->shape()) &&
144 TensorShapeUtils::IsVector(b_values_t->shape()),
163 TensorShapeUtils::IsVector(a_shape_t->shape()) &&
164 TensorShapeUtils::IsVector(b_shape_t->shape()),
H A Dsparse_fill_empty_rows_op.cc54 OP_REQUIRES(context, TensorShapeUtils::IsVector(dense_shape_t->shape()),
57 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices_t->shape()),
60 OP_REQUIRES(context, TensorShapeUtils::IsVector(values_t->shape()),
64 context, TensorShapeUtils::IsScalar(default_value_t->shape()),
213 context, TensorShapeUtils::IsVector(reverse_index_map_t->shape()),
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
H A Dbatch_features.cc47 TensorShapeUtils::IsMatrix(dense_float_feature.shape()),
77 TensorShapeUtils::IsMatrix(sparse_float_feature_indices.shape()),
81 TensorShapeUtils::IsVector(sparse_float_feature_values.shape()),
85 TensorShapeUtils::IsVector(sparse_float_feature_shape.shape()),
119 TensorShapeUtils::IsMatrix(sparse_int_feature_indices.shape()),
123 TensorShapeUtils::IsVector(sparse_int_feature_values.shape()),
126 TensorShapeUtils::IsVector(sparse_int_feature_shape.shape()),
/external/tensorflow/tensorflow/core/kernels/data/
H A Ddataset_utils.cc32 TensorShapeUtils::IsScalar(return_values[0].shape()))) {
/external/tensorflow/tensorflow/contrib/ffmpeg/
H A Dencode_audio_op.cc73 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()),
H A Ddecode_audio_op.cc119 OP_REQUIRES(context, TensorShapeUtils::IsScalar(contents_tensor.shape()),
123 OP_REQUIRES(context, TensorShapeUtils::IsScalar(file_format_tensor.shape()),
128 TensorShapeUtils::IsScalar(samples_per_second_tensor.shape()),
133 TensorShapeUtils::IsScalar(channel_count_tensor.shape()),
244 context, TensorShapeUtils::IsScalar(contents.shape()),
/external/tensorflow/tensorflow/core/framework/
H A Dtensor_shape_test.cc597 TensorShapeUtils::MakeShape(overflow, &shape).code());
619 EXPECT_TRUE(TensorShapeUtils::StartsWith(TensorShape({}), TensorShape({})));
621 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({})));
623 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({2})));
625 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({2, 3})));
626 EXPECT_TRUE(TensorShapeUtils::StartsWith(TensorShape({2, 3, 4}),
629 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({3})));
631 TensorShapeUtils::StartsWith(TensorShape({2, 3}), TensorShape({2, 4})));
632 EXPECT_FALSE(TensorShapeUtils::StartsWith(TensorShape({2, 3}),
634 EXPECT_FALSE(TensorShapeUtils
[all...]
/external/tensorflow/tensorflow/contrib/factorization/kernels/
H A Dwals_solver_ops.cc45 using tensorflow::TensorShapeUtils;
81 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(factors.shape()),
83 OP_REQUIRES(context, TensorShapeUtils::IsVector(factor_weights.shape()),
86 context, TensorShapeUtils::IsScalar(unobserved_weights.shape()),
88 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_weights.shape()),
90 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()),
92 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values.shape()),
94 OP_REQUIRES(context, TensorShapeUtils::IsScalar(input_block_size.shape()),
97 context, TensorShapeUtils::IsScalar(input_is_transpose.shape()),
H A Dmasked_matmul_ops.cc40 using tensorflow::TensorShapeUtils;
77 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(a.shape()),
79 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(b.shape()),
81 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(mask_indices.shape()),
83 OP_REQUIRES(context, TensorShapeUtils::IsScalar(transpose_a.shape()),
85 OP_REQUIRES(context, TensorShapeUtils::IsScalar(transpose_b.shape()),

Completed in 457 milliseconds

12345678