| /external/deqp-deps/amber/src/ |
| D | format.cc | 146 if (layout_ == Layout::kStd140 && (t->IsMatrix() || t->IsArray())) in NeedsPadding()
148 if (t->IsVec3() || (t->IsMatrix() && t->RowCount() == 3)) in NeedsPadding()
166 } else if (t->IsMatrix()) { in CalcArrayBaseAlignmentInBytes()
219 if (t->IsMatrix()) in CalcTypeBaseAlignmentInBytes()
248 if (type->IsStruct() || type->IsVec() || type->IsMatrix() || in AddSegmentsForType()
360 if (type->IsMatrix() && segments_.back().IsPadding()) in AddSegmentsForType()
367 if (type_->IsMatrix()) in GenerateName()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | sparse_add_op.cc | 38 TensorShapeUtils::IsMatrix(a_indices->shape()) && in Compute()
39 TensorShapeUtils::IsMatrix(b_indices->shape()), in Compute()
|
| D | matmul_op_fused.cc | 179 ctx, TensorShapeUtils::IsMatrix(a.shape()), in Compute()
183 ctx, TensorShapeUtils::IsMatrix(b.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 | sparse_reorder_op.cc | 40 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_ind.shape()), in Compute()
|
| D | sparse_split_op.cc | 38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
|
| D | sparse_tensor_dense_matmul_op.cc | 53 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(b->shape()), in Compute()
66 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a_indices->shape()), in Compute()
|
| D | sparse_tensors_map_ops.cc | 180 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
235 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
392 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(output_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_slice_op.cc | 37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
|
| D | count_ops.cc | 105 TensorShapeUtils::IsMatrix(data.shape()), in Compute()
181 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(indices.shape()), in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 61 ctx, TensorShapeUtils::IsMatrix(a_shape), in Compile()
65 ctx, TensorShapeUtils::IsMatrix(b_shape), in Compile()
|
| D | mirror_pad_op.cc | 80 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
179 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, 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 | reduce_window_op.cc | 104 TensorShapeUtils::IsMatrix(padding_shape) && in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op.cc | 53 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(a.shape()), in Compute()
55 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(b.shape()), in Compute()
|
| D | mkl_matmul_op_fused.cc | 76 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(src_tf_shape), in Compute()
78 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(weight_tf_shape), in Compute()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.cc | 26 DCHECK(TensorShapeUtils::IsMatrix(ix.shape())); in UnsafeGetDimsFromIx()
31 if (!TensorShapeUtils::IsMatrix(ix.shape())) { in GetDimsFromIx()
|