| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | batch_features.cc | 48 TensorShapeUtils::IsMatrix(dense_float_feature.shape()), in Initialize()
78 TensorShapeUtils::IsMatrix(sparse_float_feature_indices.shape()), in Initialize()
122 TensorShapeUtils::IsMatrix(sparse_int_feature_indices.shape()), in Initialize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | linalg_ops_common.cc | 38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[0]), in ValidateSingleMatrix()
60 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[0]), in ValidateSolver()
62 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[1]), in ValidateSolver()
80 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_matrix_shapes[1]), in ValidateSquareSolver()
|
| D | sparse_add_grad_op.cc | 39 TensorShapeUtils::IsMatrix(a_indices->shape()) && in Compute()
40 TensorShapeUtils::IsMatrix(b_indices->shape()) && in Compute()
41 TensorShapeUtils::IsMatrix(sum_indices->shape()), in Compute()
|
| D | sparse_slice_grad_op.cc | 37 TensorShapeUtils::IsMatrix(input_indices->shape()) && in Compute()
38 TensorShapeUtils::IsMatrix(output_indices->shape()), in Compute()
|
| D | tensor_flag_utils.cc | 35 if (!TensorShapeUtils::IsMatrix(config.shape())) { in ValidateSparseMatrixShardingConfig()
99 if (!TensorShapeUtils::IsMatrix(config.shape())) { in ValidateScalarQuantityShardingConfig()
|
| D | mkl_matmul_op.cc | 57 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a.shape()), in Compute()
59 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(b.shape()), in Compute()
|
| D | sparse_add_op.cc | 38 TensorShapeUtils::IsMatrix(a_indices->shape()) && in Compute()
39 TensorShapeUtils::IsMatrix(b_indices->shape()), in Compute()
|
| D | sparse_sparse_binary_op_shared.cc | 136 TensorShapeUtils::IsMatrix(a_indices_t->shape()) && in Compute()
137 TensorShapeUtils::IsMatrix(b_indices_t->shape()), in Compute()
|
| D | edit_distance_op.cc | 43 if (!TensorShapeUtils::IsMatrix(hypothesis_indices.shape())) in ValidateShapes()
47 if (!TensorShapeUtils::IsMatrix(truth_indices.shape())) in ValidateShapes()
|
| D | quantized_matmul_op.cc | 97 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(a.shape()), in Compute()
99 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(b.shape()), in Compute()
|
| D | serialize_sparse_op.cc | 62 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
168 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
|
| D | sparse_reorder_op.cc | 40 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_ind.shape()), in Compute()
|
| D | sparse_tensor_dense_matmul_op.cc | 52 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(b->shape()), in Compute()
65 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a_indices->shape()), in Compute()
|
| D | clustering_ops.cc | 89 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(points_tensor.shape()), in Compute()
295 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(points_tensor.shape()), in Compute()
297 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(centers_tensor.shape()), in Compute()
|
| D | sparse_tensors_map_ops.cc | 179 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
234 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
391 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(output_indices.shape()), in Compute()
|
| D | sparse_split_op.cc | 38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
|
| D | sparse_slice_op.cc | 37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | masked_matmul_ops.cc | 77 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(a.shape()), in Compute()
79 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(b.shape()), in Compute()
81 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(mask_indices.shape()), in Compute()
|
| D | wals_solver_ops.cc | 83 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(factors.shape()), in Compute()
92 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 58 ctx, TensorShapeUtils::IsMatrix(a_shape), in Compile()
62 ctx, TensorShapeUtils::IsMatrix(b_shape), in Compile()
|
| D | softmax_op.cc | 155 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_shape), in Compile()
183 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_shape), in Compile()
|
| D | pad_op.cc | 38 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
|
| D | xla_conv_op.cc | 61 TensorShapeUtils::IsMatrix(padding_shape) && in Compile()
|
| D | mirror_pad_op.cc | 78 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| D | encode_audio_op.cc | 73 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(contents.shape()), in Compute()
178 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(contents.shape()), in Compute()
|