| /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()),
122 TensorShapeUtils::IsMatrix(sparse_int_feature_indices.shape()),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_add_grad_op.cc | 39 TensorShapeUtils::IsMatrix(a_indices->shape()) &&
40 TensorShapeUtils::IsMatrix(b_indices->shape()) &&
41 TensorShapeUtils::IsMatrix(sum_indices->shape()),
|
| linalg_ops_common.cc | 38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[0]),
60 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[0]),
62 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[1]),
80 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[1]),
|
| sparse_slice_grad_op.cc | 37 TensorShapeUtils::IsMatrix(input_indices->shape()) &&
38 TensorShapeUtils::IsMatrix(output_indices->shape()),
|
| tensor_flag_utils.cc | 35 if (!TensorShapeUtils::IsMatrix(config.shape())) {
99 if (!TensorShapeUtils::IsMatrix(config.shape())) {
|
| sparse_reorder_op.cc | 40 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_ind.shape()),
|
| sparse_slice_op.cc | 37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()),
|
| sparse_split_op.cc | 38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()),
|
| mkl_matmul_op.cc | 57 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a.shape()),
59 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(b.shape()),
|
| quantized_matmul_op.cc | 97 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(a.shape()),
99 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(b.shape()),
|
| sparse_add_op.cc | 38 TensorShapeUtils::IsMatrix(a_indices->shape()) &&
39 TensorShapeUtils::IsMatrix(b_indices->shape()),
|
| sparse_sparse_binary_op_shared.cc | 136 TensorShapeUtils::IsMatrix(a_indices_t->shape()) &&
137 TensorShapeUtils::IsMatrix(b_indices_t->shape()),
|
| reshape_util.cc | 37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices_in.shape()),
|
| sparse_dense_binary_op_shared.cc | 69 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(indices_t->shape()),
|
| sparse_softmax_op.cc | 49 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices_t->shape()),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| matmul_op.cc | 58 ctx, TensorShapeUtils::IsMatrix(a_shape),
62 ctx, TensorShapeUtils::IsMatrix(b_shape),
|
| pad_op.cc | 38 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2,
|
| xla_conv_op.cc | 61 TensorShapeUtils::IsMatrix(padding_shape) &&
|
| mirror_pad_op.cc | 78 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2,
|
| categorical_op.cc | 46 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_shape),
|
| reduce_window_op.cc | 105 TensorShapeUtils::IsMatrix(padding_shape) &&
|
| xla_select_and_scatter_op.cc | 112 TensorShapeUtils::IsMatrix(padding_shape) &&
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| encode_audio_op.cc | 73 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(contents.shape()),
178 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(contents.shape()),
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| masked_matmul_ops.cc | 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()),
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| image_ops.cc | 77 (TensorShapeUtils::IsMatrix(transform_t.shape()) &&
|
