Home
last modified time | relevance | path

Searched refs:GetDimension (Results 1 – 25 of 58) sorted by relevance

123

/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_conversion.cpp47 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
73 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
101 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
128 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
154 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
181 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
237 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
409 _.GetBitWidth(result_type) * _.GetDimension(result_type); in ConversionPass()
411 _.GetBitWidth(input_type) * _.GetDimension(input_type); in ConversionPass()
Dvalidate_extensions.cpp169 const uint32_t result_type_dimension = _.GetDimension(result_type); in ValidateExtInst()
180 if (result_type_dimension != _.GetDimension(operand_type)) { in ValidateExtInst()
395 if (_.GetDimension(result_type) != _.GetDimension(exp_data_type)) { in ValidateExtInst()
428 if (_.GetDimension(result_type) != _.GetDimension(exp_type)) { in ValidateExtInst()
449 _.GetDimension(result_types[0]) != in ValidateExtInst()
450 _.GetDimension(result_types[1])) { in ValidateExtInst()
482 if (!_.IsFloatVectorType(v_type) || _.GetDimension(v_type) != 4 || in ValidateExtInst()
502 if (!_.IsFloatVectorType(v_type) || _.GetDimension(v_type) != 2 || in ValidateExtInst()
520 if (!_.IsIntVectorType(v_type) || _.GetDimension(v_type) != 2 || in ValidateExtInst()
532 _.GetDimension(result_type) != 4 || in ValidateExtInst()
[all …]
Dvalidate_logicals.cpp66 if (_.GetDimension(result_type) != _.GetDimension(operand_type)) in LogicalsPass()
102 if (_.GetDimension(result_type) != _.GetDimension(left_operand_type)) in LogicalsPass()
194 if (_.GetDimension(condition_type) != dimension) in LogicalsPass()
231 if (_.GetDimension(result_type) != _.GetDimension(left_type)) in LogicalsPass()
242 if (_.GetDimension(result_type) != _.GetDimension(right_type)) in LogicalsPass()
Dvalidate_bitwise.cpp41 const uint32_t result_dimension = _.GetDimension(result_type); in BitwisePass()
51 if (_.GetDimension(base_type) != result_dimension) in BitwisePass()
67 if (_.GetDimension(shift_type) != result_dimension) in BitwisePass()
83 const uint32_t result_dimension = _.GetDimension(result_type); in BitwisePass()
96 if (_.GetDimension(type_id) != result_dimension) in BitwisePass()
200 const uint32_t base_dimension = _.GetDimension(base_type); in BitwisePass()
201 const uint32_t result_dimension = _.GetDimension(result_type); in BitwisePass()
Dvalidate_arithmetics.cpp90 const uint32_t dimension = _.GetDimension(result_type); in ArithmeticsPass()
103 if (_.GetDimension(type_id) != dimension) in ArithmeticsPass()
143 const uint32_t num_components = _.GetDimension(type_id); in ArithmeticsPass()
238 if (matrix_num_cols != _.GetDimension(result_type)) in ArithmeticsPass()
243 if (matrix_num_rows != _.GetDimension(vector_type_id)) in ArithmeticsPass()
287 if (matrix_num_cols != _.GetDimension(vector_type_id)) in ArithmeticsPass()
390 if (res_num_cols != _.GetDimension(right_type_id)) in ArithmeticsPass()
Dvalidate_image.cpp333 const uint32_t dx_size = _.GetDimension(dx_type_id); in ValidateImageOperands()
334 const uint32_t dy_size = _.GetDimension(dy_type_id); in ValidateImageOperands()
374 const uint32_t offset_size = _.GetDimension(type_id); in ValidateImageOperands()
397 const uint32_t offset_size = _.GetDimension(type_id); in ValidateImageOperands()
442 _.GetDimension(component_type) != 2) { in ValidateImageOperands()
918 const uint32_t actual_coord_size = _.GetDimension(coord_type); in ValidateImageTexelPointer()
957 if (_.GetDimension(actual_result_type) != 4) { in ValidateImageLod()
1004 const uint32_t actual_coord_size = _.GetDimension(coord_type); in ValidateImageLod()
1066 const uint32_t actual_coord_size = _.GetDimension(coord_type); in ValidateImageDrefLod()
1106 if (_.GetDimension(actual_result_type) != 4) { in ValidateImageFetch()
[all …]
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_extensions.cpp169 const uint32_t result_type_dimension = _.GetDimension(result_type); in ValidateExtInst()
180 if (result_type_dimension != _.GetDimension(operand_type)) { in ValidateExtInst()
395 if (_.GetDimension(result_type) != _.GetDimension(exp_data_type)) { in ValidateExtInst()
428 if (_.GetDimension(result_type) != _.GetDimension(exp_type)) { in ValidateExtInst()
449 _.GetDimension(result_types[0]) != in ValidateExtInst()
450 _.GetDimension(result_types[1])) { in ValidateExtInst()
482 if (!_.IsFloatVectorType(v_type) || _.GetDimension(v_type) != 4 || in ValidateExtInst()
502 if (!_.IsFloatVectorType(v_type) || _.GetDimension(v_type) != 2 || in ValidateExtInst()
520 if (!_.IsIntVectorType(v_type) || _.GetDimension(v_type) != 2 || in ValidateExtInst()
532 _.GetDimension(result_type) != 4 || in ValidateExtInst()
[all …]
Dvalidate_conversion.cpp47 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
73 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
101 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
128 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
154 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
181 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
254 if (_.GetDimension(result_type) != _.GetDimension(input_type)) in ConversionPass()
443 _.GetBitWidth(result_type) * _.GetDimension(result_type); in ConversionPass()
445 _.GetBitWidth(input_type) * _.GetDimension(input_type); in ConversionPass()
Dvalidate_logicals.cpp66 if (_.GetDimension(result_type) != _.GetDimension(operand_type)) in LogicalsPass()
102 if (_.GetDimension(result_type) != _.GetDimension(left_operand_type)) in LogicalsPass()
195 if (_.GetDimension(condition_type) != dimension) in LogicalsPass()
232 if (_.GetDimension(result_type) != _.GetDimension(left_type)) in LogicalsPass()
243 if (_.GetDimension(result_type) != _.GetDimension(right_type)) in LogicalsPass()
Dvalidate_bitwise.cpp41 const uint32_t result_dimension = _.GetDimension(result_type); in BitwisePass()
51 if (_.GetDimension(base_type) != result_dimension) in BitwisePass()
67 if (_.GetDimension(shift_type) != result_dimension) in BitwisePass()
83 const uint32_t result_dimension = _.GetDimension(result_type); in BitwisePass()
96 if (_.GetDimension(type_id) != result_dimension) in BitwisePass()
200 const uint32_t base_dimension = _.GetDimension(base_type); in BitwisePass()
201 const uint32_t result_dimension = _.GetDimension(result_type); in BitwisePass()
Dvalidate_arithmetics.cpp90 const uint32_t dimension = _.GetDimension(result_type); in ArithmeticsPass()
103 if (_.GetDimension(type_id) != dimension) in ArithmeticsPass()
143 const uint32_t num_components = _.GetDimension(type_id); in ArithmeticsPass()
238 if (matrix_num_cols != _.GetDimension(result_type)) in ArithmeticsPass()
243 if (matrix_num_rows != _.GetDimension(vector_type_id)) in ArithmeticsPass()
287 if (matrix_num_cols != _.GetDimension(vector_type_id)) in ArithmeticsPass()
390 if (res_num_cols != _.GetDimension(right_type_id)) in ArithmeticsPass()
Dvalidate_image.cpp333 const uint32_t dx_size = _.GetDimension(dx_type_id); in ValidateImageOperands()
334 const uint32_t dy_size = _.GetDimension(dy_type_id); in ValidateImageOperands()
374 const uint32_t offset_size = _.GetDimension(type_id); in ValidateImageOperands()
397 const uint32_t offset_size = _.GetDimension(type_id); in ValidateImageOperands()
442 _.GetDimension(component_type) != 2) { in ValidateImageOperands()
918 const uint32_t actual_coord_size = _.GetDimension(coord_type); in ValidateImageTexelPointer()
957 if (_.GetDimension(actual_result_type) != 4) { in ValidateImageLod()
1004 const uint32_t actual_coord_size = _.GetDimension(coord_type); in ValidateImageLod()
1066 const uint32_t actual_coord_size = _.GetDimension(coord_type); in ValidateImageDrefLod()
1106 if (_.GetDimension(actual_result_type) != 4) { in ValidateImageFetch()
[all …]
Dvalidate_non_uniform.cpp45 _.GetDimension(value_type) != 4) { in ValidateGroupNonUniformBallotBitCount()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc44 int64 n = ShapeUtil::GetDimension(shape, -1); in DiagonalBlocks()
78 slice_sizes[i] = ShapeUtil::GetDimension(shape, i); in DiagonalBlocks()
132 int64 block_size = ShapeUtil::GetDimension(shape, -1); in InvertDiagonalBlocks()
260 int64 block_size = ShapeUtil::GetDimension(blocks_shape, -1); in SolveWithInvertedDiagonalBlocks()
264 int64 n = ShapeUtil::GetDimension(a_shape, -1); in SolveWithInvertedDiagonalBlocks()
267 int64 m = ShapeUtil::GetDimension(b_shape, m_dim); in SolveWithInvertedDiagonalBlocks()
384 if (ShapeUtil::GetDimension(a_shape, -1) != in BuildTriangularSolve()
385 ShapeUtil::GetDimension(a_shape, -2)) { in BuildTriangularSolve()
391 const int64 m = ShapeUtil::GetDimension(b_shape, -2); in BuildTriangularSolve()
392 const int64 n = ShapeUtil::GetDimension(b_shape, -1); in BuildTriangularSolve()
[all …]
Dshape_inference.cc208 ShapeUtil::GetDimension(base_shape, i), dim.base_dilation()); in InferWindowOutputShape()
1250 if (ShapeUtil::GetDimension(offset_shape, 0) != feature_count) { in InferBatchNormTrainingShape()
1255 ShapeUtil::GetDimension(offset_shape, 0), feature_count); in InferBatchNormTrainingShape()
1258 if (ShapeUtil::GetDimension(scale_shape, 0) != feature_count) { in InferBatchNormTrainingShape()
1263 ShapeUtil::GetDimension(scale_shape, 0), feature_count); in InferBatchNormTrainingShape()
1380 if (ShapeUtil::GetDimension(offset_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
1385 ShapeUtil::GetDimension(offset_shape, 0), feature_count); in InferBatchNormInferenceShape()
1388 if (ShapeUtil::GetDimension(scale_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
1393 ShapeUtil::GetDimension(scale_shape, 0), feature_count); in InferBatchNormInferenceShape()
1396 if (ShapeUtil::GetDimension(mean_shape, 0) != feature_count) { in InferBatchNormInferenceShape()
[all …]
Dcholesky_expander.cc59 const int64 n = ShapeUtil::GetDimension(a_shape, -1); in CholeskyUnblocked()
150 const int64 n = ShapeUtil::GetDimension(a_shape, -1); in BuildCholesky()
151 if (n != ShapeUtil::GetDimension(a_shape, -2)) { in BuildCholesky()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd.cc119 const int64 n = ShapeUtil::GetDimension(a_shape, -1); in HouseRow()
126 batch_dims[k] = ShapeUtil::GetDimension(a_shape, k); in HouseRow()
185 const int64 m = ShapeUtil::GetDimension(a_shape, -2); in HouseCol()
192 batch_dims[k] = ShapeUtil::GetDimension(a_shape, k); in HouseCol()
266 batch_dims[i] = ShapeUtil::GetDimension(a_shape, i); in HouseHolderBidiagonalization()
268 const int64 m = ShapeUtil::GetDimension(a_shape, -2); in HouseHolderBidiagonalization()
269 const int64 n = ShapeUtil::GetDimension(a_shape, -1); in HouseHolderBidiagonalization()
471 batch_dims[i] = ShapeUtil::GetDimension(d_shape, i); in OneSidedJacobiUpdate()
473 const int64 m = ShapeUtil::GetDimension(d_shape, -2); in OneSidedJacobiUpdate()
474 const int64 n = ShapeUtil::GetDimension(d_shape, -1); in OneSidedJacobiUpdate()
[all …]
Dqr.cc164 const int64 m = ShapeUtil::GetDimension(a_shape, -2); in QRBlock()
165 const int64 n = ShapeUtil::GetDimension(a_shape, -1); in QRBlock()
170 batch_dims[i] = ShapeUtil::GetDimension(a_shape, i); in QRBlock()
335 const int64 m = ShapeUtil::GetDimension(a_shape, -2); in QRDecomposition()
336 const int64 n = ShapeUtil::GetDimension(a_shape, -1); in QRDecomposition()
347 batch_dims[i] = ShapeUtil::GetDimension(a_shape, i); in QRDecomposition()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_lrn_op.cc109 } else if (!src_dnn_shape.IsMklChannelDim(src_dnn_shape.GetDimension() - in Compute()
293 OP_REQUIRES(context, src_dnn_shape.GetDimension() == 4, in SanityCheckInputs()
362 input_grad_dnn_shape.GetDimension() - 1) && in Compute()
364 orig_input_dnn_shape.GetDimension() - 1) && in Compute()
366 orig_output_dnn_shape.GetDimension() - 1); in Compute()
618 OP_REQUIRES(context, in_grads_dnn_shape.GetDimension() == 4, in SanityCheckInputs()
628 OP_REQUIRES(context, in_image_dnn_shape.GetDimension() == 4, in SanityCheckInputs()
638 OP_REQUIRES(context, out_image_dnn_shape.GetDimension() == 4, in SanityCheckInputs()
Dmkl_pooling_ops_common.cc275 depth = mklInputShape->GetDimension('C'); in Init()
276 tensor_in_cols = mklInputShape->GetDimension('W'); in Init()
277 tensor_in_rows = mklInputShape->GetDimension('H'); in Init()
278 tensor_in_batch = mklInputShape->GetDimension('N'); in Init()
Dmkl_avgpooling_op.cc77 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension(); in Compute()
264 mkl_context.params.in_dim = mkl_context.out_backprop_shape.GetDimension(); in Compute()
374 context, out_backprop_shape.GetDimension() == 4, in MklCreateLayoutsAndPrimitives()
685 original_input_mkl_shape.GetDimension() == 1 && in SanityCheckInputs()
698 context, input_gradient_mkl_shape.GetDimension() == 4, in SanityCheckInputs()
Dmkl_input_conversion_op.cc127 mkl_output_mkl_shape.SetTfLayout(input_shape_0.GetDimension(), in Compute()
225 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h241 size_t GetDimension() const { return dimension_; } in GetDimension() function
512 inline size_t GetDimension(char dimension) const { in GetDimension() function
514 CHECK(index >= 0 && index < this->GetDimension()) in GetDimension()
521 CHECK(index >= 0 && index < this->GetDimension()) in GetDimension3D()
560 inline size_t GetDimension() const { return data_.dimension_; } in GetDimension() function
808 for (size_t j = 0; j < mkl_shape.GetDimension(); j++) { in ConvertMklToTF()
967 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension())); in AllocateOutputSetMklShape()
1002 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension())); in AllocateOutputSetMklShape()
1092 size_t tf_dim = mkl_shape.GetDimension(); in MklSizesToTFSizes()
1134 CHECK(index >= 0 && index < mkl_shape.GetDimension()) in GetMklTensorDim()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatchtospace_op.cc43 block_rank == xla::ShapeUtil::GetDimension(crops.shape(), 0) && in BatchToSpace()
44 2 == xla::ShapeUtil::GetDimension(crops.shape(), 1), in BatchToSpace()
Dspacetobatch_op.cc43 block_rank == xla::ShapeUtil::GetDimension(paddings.shape(), 0) && in SpaceToBatch()
44 2 == xla::ShapeUtil::GetDimension(paddings.shape(), 1), in SpaceToBatch()

123