Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 202 for pushBack (0.29 sec)

  1. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc

            launch_results.push_back(result);
            launch_result_types.push_back(result.getType());
          }
        }
      }
    
      before ? builder->setInsertionPoint(op) : builder->setInsertionPointAfter(op);
      auto launch = builder->create<mlir::tf_device::LaunchOp>(
          op->getLoc(), builder->getStringAttr(host_device), launch_result_types);
      launch.getBody().push_back(launch_block);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.6K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc

      std::vector<TFE_TensorHandle*> compute_fn_inputs;
      TFE_TensorHandle* input_a = TestScalarTensorHandle(ctx, 2.0f);
      TFE_TensorHandle* input_b = TestScalarTensorHandle(ctx, 1.0f);
      compute_fn_inputs.push_back(input_a);
      compute_fn_inputs.push_back(input_b);
    
      TFE_Op* compute_fn_op = TF_ConcreteFunctionMakeCallOp(
          compute_fn, compute_fn_inputs.data(), compute_fn_inputs.size(), status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 08:08:45 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

        // 1.0e-6.
        if (max - min > kNearZeroTolerance) {
          effective_mins.push_back(min);
          effective_maxs.push_back(max);
        } else {
          effective_mins.push_back(std::min(min, -kSmallestHalfRange));
          effective_maxs.push_back(std::max(max, kSmallestHalfRange));
        }
      }
    }
    
    // Sets the min / max, scale and zero_points from the fake quant num_bits
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc

            return rewriter.notifyMatchFailure(
                second, "non-concat dims have mismatching slice bounds");
        }
    
        new_start.push_back(first_start);
        new_limit.push_back(second_limit);
        new_slice_shape.push_back(second_limit - first_start);
      }
    
      auto new_slice = rewriter.create<mhlo::SliceOp>(
          FusedLoc::get(first->getContext(), {first.getLoc(), second.getLoc()}),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 26.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc

        for (auto& execute : executes)
          island_operands.push_back(execute.getControl());
    
        builder.setInsertionPoint(island_op);
        auto island_sink = builder.create<tf_executor::IslandOp>(
            island_op.getLoc(), llvm::ArrayRef<Type>{},
            island_op.getControl().getType(), island_operands);
        island_sink.getBody().push_back(new Block);
        builder.setInsertionPointToEnd(&island_sink.GetBody());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 19 19:47:16 UTC 2023
    - 11.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

      // `replicate`.
      llvm::SmallVector<Value, 8> reformat_operands;
      for (const auto& entry : execute_arg_to_outer_args) {
        reformat_operands.push_back(execute.getArgs()[entry.first]);
      }
      reformat_operands.push_back(compile_launch.getResult(1));
      reformat_operands.push_back(replicate.GetBody().getArgument(
          replicate.GetNumReplicatedBlockArguments() - 1));
      builder.setInsertionPoint(execute_launch);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_function.cc

        for (int i = 0; i < noutputs; ++i) {
          output_names_vec.push_back(string(output_names[i]));
        }
      }
    
      // Process control output names.
      std::vector<string> control_output_names_vec;
      if (control_output_names) {
        control_output_names_vec.reserve(ncontrol_outputs);
        for (int i = 0; i < ncontrol_outputs; ++i) {
          control_output_names_vec.push_back(string(control_output_names[i]));
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

          if (is_guaranteed_constant(*n)) {
            return errors::InvalidArgument(
                "Guaranteed constants are not supported (", n->name(), ")");
          }
          args.push_back(n);
        } else if (n->type_string() == "_Retval") {
          retvals.push_back(n);
        }
      }
    
      if (std::find(args.begin(), args.end(), nullptr) != args.end()) {
        return errors::InvalidArgument("Missing or non-consecutive arguments");
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

            for (int i = 0; i < repeats; ++i)
              state_->types.push_back(UnrankedTensorType::get(type_attr.getType()));
          } else if (output_arg.type() != tensorflow::DT_INVALID) {
            for (int i = 0; i < repeats; ++i) {
              Type type;
              TF_RETURN_IF_ERROR(
                  ConvertDataType(output_arg.type(), builder, &type));
              state_->types.push_back(type);
            }
          } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  10. tensorflow/cc/framework/cc_op_gen_util.cc

        CHECK_EQ(arg.name(), api_def_arg.name());
    
        bool is_list = ArgIsList(arg);
        output_types.push_back(
            strings::StrCat("::tensorflow::", is_list ? "OutputList" : "Output"));
        output_names.push_back(AvoidCPPKeywords(api_def_arg.rename_to()));
        is_list_output.push_back(is_list);
      }
    
      strings::StrAppend(&comment, "\nReturns:\n");
      if (graph_op_def.output_arg_size() == 0) {  // No outputs.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
Back to top