Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for _contains_op_with_name_and_attribute (0.5 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

        Args:
          func: A FunctionDef object.
    
        Returns:
          True iff `func` is composte function.
        """
        return func.signature.name.startswith('composite_')
    
      def _contains_op_with_name_and_attribute(
          self,
          nodes: Iterable[node_def_pb2.NodeDef],
          op_name: str,
          attr_name: str,
          attr_val: _AttrValType,
          node_name: str = '',
      ) -> bool:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

          if alias == 'conv_func':
            for func in meta_graph_def.graph_def.library.function:
              if func.signature.name == func_name:
                self.assertTrue(
                    self._contains_op_with_name_and_attribute(
                        func.node_def,
                        op_name='XlaConvV2',
                        attr_name='',
                        attr_val=None,
                    )
                )
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
Back to top