Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_Op (0.06 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

    }
    
    def TF_FlushSummaryWriterOp : TF_Op<"FlushSummaryWriter", []> {
      let summary = "Flushes the writer's unwritten events.";
    
      let description = [{
    writer: A handle to the summary writer resource.
      }];
    
      let arguments = (ins
        Arg<TF_ResourceTensor, "", [TF_SummaryWrite]>:$writer
      );
    
      let results = (outs);
    }
    
    def TF_ImportEventOp : TF_Op<"ImportEvent", []> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        // * The 'dilations' attribute is optional, but it must be of the form
        //   [1, X, Y, 1] if exists.
    
        TFConvOpType tf_op = cast<TFConvOpType>(op);
        if (!TFTypeIsFloat32Tensor(tf_op.getInput()) &&
            !(allow_bf16_and_f16_type_legalization_ &&
              TFTypeIsBFloat16OrHalfTensor(tf_op.getInput())))
          return failure();
    
        if (!TFDataFormatIsNHWC(op)) return failure();
    
        IntegerAttr height, width;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
Back to top