Home
last modified time | relevance | path

Searched refs:DimensionType (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc58 string DebugString(const nvinfer1::DimensionType type) { in DebugString()
60 case nvinfer1::DimensionType::kSPATIAL: in DebugString()
62 case nvinfer1::DimensionType::kCHANNEL: in DebugString()
64 case nvinfer1::DimensionType::kINDEX: in DebugString()
66 case nvinfer1::DimensionType::kSEQUENCE: in DebugString()
Dutils.h103 string DebugString(const nvinfer1::DimensionType type);
Dconvert_nodes.cc4266 dims->type[0] = nvinfer1::DimensionType::kSPATIAL; in GetTensorDimsWithProtoShape()
/external/deqp/modules/gles31/functional/
Des31fFramebufferDefaultStateQueryTests.cpp101 enum DimensionType enum in deqp::gles31::Functional::__anon0bc859400111::FramebufferDimensionTest
109 …FramebufferDimensionTest (Context& context, QueryType verifier, DimensionType dimension, const cha…
114 const DimensionType m_dimension;
117 …Test::FramebufferDimensionTest (Context& context, QueryType verifier, DimensionType dimension, con… in FramebufferDimensionTest()
/external/tensorflow/tensorflow/core/kernels/linalg/
Deinsum_op_impl.h69 enum DimensionType { enum
92 static DimensionType GetDimensionType(bool is_removed, bool is_unique) { in GetDimensionType()
129 std::vector<DimensionType>* label_types, in ParseEquation()
222 Labels* output_labels, std::vector<DimensionType>* label_types, in ProcessDimensions()
438 const Labels& labels, const std::vector<DimensionType>& label_types) { in ShouldSwapFreeAndContract()
455 const std::vector<DimensionType>& label_types, in ReduceOperand()
607 std::vector<EinsumHelper::DimensionType> label_types(label_types_); in Compute()
734 std::vector<EinsumHelper::DimensionType> label_types_;
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_op_interfaces.td46 "DimensionType", "dimension type", [TFL_DT_Dense, TFL_DT_SparseCSR]>;
/external/tensorflow/tensorflow/lite/schema/
Dschema_v3a.fbs113 enum DimensionType : byte {
141 format:DimensionType;
143 // - If format is DimensionType.DENSE then we use the dense_size field to
146 // - If format is DimensionType.SPARSE_CSR then we use array_segments and
Dschema.fbs117 enum DimensionType : byte {
145 format:DimensionType;
147 // - If format is DimensionType.DENSE then we use the dense_size field to
150 // - If format is DimensionType.SPARSE_CSR then we use array_segments and
Dschema_generated.h546 enum DimensionType {
553 inline const DimensionType (&EnumValuesDimensionType())[2] {
554 static const DimensionType values[] = {
570 inline const char *EnumNameDimensionType(DimensionType e) {
3447 tflite::DimensionType format;
3467 tflite::DimensionType format() const {
3468 return static_cast<tflite::DimensionType>(GetField<int8_t>(VT_FORMAT, 0));
3549 void add_format(tflite::DimensionType format) {
3581 tflite::DimensionType format = tflite::DimensionType_DENSE,
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dtest_schema.fbs110 enum DimensionType : byte {
138 format:DimensionType;
140 // - If format is DimensionType.DENSE then we use the dense_size field to
143 // - If format is DimensionType.SPARSE_CSR then we use array_segments and