/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | 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),
|
H A D | select_op.cc | 50 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 D | training_ops.cc | 40 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 D | bias_ops.cc | 44 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 D | matrix_band_part_op.cc | 32 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 D | pad_op.cc | 36 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 D | 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)) {
|
H A D | 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()),
|
H A D | 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()),
|
H A D | matching_files_op.cc | 38 TensorShapeUtils::IsScalar(patterns_t->shape()) || 39 TensorShapeUtils::IsVector(patterns_t->shape()),
|
H A D | 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()),
|
H A D | 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()),
|
H A D | 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()),
|
H A D | 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())) {
|
H A D | 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(
|
H A D | linalg_ops_common.cc | 38 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 D | sparse_sparse_binary_op_shared.cc | 136 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 D | sparse_fill_empty_rows_op.cc | 54 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 D | batch_features.cc | 47 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 D | dataset_utils.cc | 32 TensorShapeUtils::IsScalar(return_values[0].shape()))) {
|
/external/tensorflow/tensorflow/contrib/ffmpeg/ |
H A D | 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()),
|
H A D | decode_audio_op.cc | 119 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 D | tensor_shape_test.cc | 597 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 D | wals_solver_ops.cc | 45 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 D | masked_matmul_ops.cc | 40 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()),
|