Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 22 for type_attr (0.58 sec)

  1. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

      DataType dtype;
      TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype));
      AttrValue type_attr;
      type_attr.set_type(dtype);
      (*node_def->mutable_attr())["T"] = type_attr;
    
      AttrValue index_attr;
      index_attr.set_i(index);
      (*node_def->mutable_attr())["index"] = index_attr;
    
      if (auto device_attr =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

        for (auto type_and_name :
             llvm::zip(intermediate_types, kIntermediateNames)) {
          mlir::TypeAttr type_attr =
              mlir::TypeAttr::get(std::get<0>(type_and_name));
          auto named_attr =
              builder.getNamedAttr(std::get<1>(type_and_name), type_attr);
          op_state.addAttribute(named_attr.getName(), named_attr.getValue());
        }
      }
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

          return false;
        }
      }
      return true;
    }
    
    // Returns the tensor type created from the `shape_attr` and `type_attr`
    // attributes.
    Type GetType(Attribute shape_attr, Attribute type_attr) {
      auto shape = mlir::cast<tf_type::ShapeAttr>(shape_attr);
      auto type = mlir::cast<TypeAttr>(type_attr);
      if (shape.hasRank())
        return tensorflow::GetTypeFromTFTensorShape(shape.getShape(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

        type_attr = attr_value_pb2.AttrValue(type=types_pb2.DT_QINT8)
        if quantize:
          self.assertTrue(
              self._contains_op(output_graphdef, 'Const', 'dtype', type_attr)
          )
        else:
          self.assertFalse(
              self._contains_op(output_graphdef, 'Const', 'dtype', type_attr)
          )
    
      @parameterized.named_parameters(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        // determined while going through quantization passes.
        OptionalAttr<TypeAttr>:$input_to_input_intermediate,
        OptionalAttr<TypeAttr>:$input_to_forget_intermediate,
        OptionalAttr<TypeAttr>:$input_to_cell_intermediate,
        OptionalAttr<TypeAttr>:$input_to_output_intermediate,
        OptionalAttr<TypeAttr>:$effective_hidden_scale_intermediate
      );
    
      let results = (outs AnyTensor:$output);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

      if (tensorflow::IsRefType(dtype)) dtype = tensorflow::RemoveRefType(dtype);
      value->set_type(dtype);
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::TypeAttr& type, bool remove_ref_type,
                            AttrValue* value) {
      return ConvertAttribute(type.getValue(), remove_ref_type, value);
    }
    
    Status ConvertAttribute(const mlir::UnitAttr& attr, AttrValue* value) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc

        builder.create<tf_saved_model::GlobalTensorOp>(
            NameLoc::get(builder.getStringAttr(name.str())),
            builder.getStringAttr(name), tensor_attr,
            TypeAttr::get(tensor_attr.getType()), builder.getUnitAttr());
      }
    
      return success();
    }
    
    }  // namespace
    
    LogicalResult LiftVariables(ModuleOp module, Session* session) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 7.9K bytes
    - Viewed (0)
  8. src/cmd/internal/obj/ppc64/obj9.go

    	}
    
    	var source *obj.Addr
    	if p.From.Name == obj.NAME_EXTERN || p.From.Name == obj.NAME_STATIC {
    		if p.From.Type == obj.TYPE_ADDR {
    			if p.As == ADWORD {
    				// ADWORD $sym doesn't need TOC anchor
    				return
    			}
    			if p.As != AMOVD {
    				c.ctxt.Diag("do not know how to handle TYPE_ADDR in %v", p)
    				return
    			}
    			if p.To.Type != obj.TYPE_REG {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 22 18:17:17 UTC 2024
    - 40.8K bytes
    - Viewed (0)
  9. src/cmd/compile/internal/ppc64/ssa.go

    			if v.AuxInt != 0 || v.Args[0].Reg() != v.Reg() {
    				p := s.Prog(ppc64.AMOVD)
    				p.From.Type = obj.TYPE_ADDR
    				p.From.Reg = v.Args[0].Reg()
    				p.From.Offset = v.AuxInt
    				p.To.Type = obj.TYPE_REG
    				p.To.Reg = v.Reg()
    			}
    
    		case *obj.LSym, ir.Node:
    			p := s.Prog(ppc64.AMOVD)
    			p.From.Type = obj.TYPE_ADDR
    			p.From.Reg = v.Args[0].Reg()
    			p.To.Type = obj.TYPE_REG
    			p.To.Reg = v.Reg()
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 22 19:59:38 UTC 2024
    - 55.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

                res_types, res_attrs)))
          return failure();
    
        // Update the function type which has excluded the control args.
        func->setAttr("function_type", TypeAttr::get(rewriter.getFunctionType(
                                           arg_types, res_types)));
    
        // Update arg/result attributes.
        func.setAllArgAttrs(arg_attrs);
        func.setAllResultAttrs(res_attrs);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
Back to top