Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 442 for pushBack (0.18 sec)

  1. tensorflow/compiler/mlir/tfr/utils/utils.cc

          return {};
        }
        if (tf_op_name[i] == '.') {
          compose_func_name.push_back('_');
        } else if (tf_op_name[i] >= 'A' && tf_op_name[i] <= 'Z') {
          compose_func_name.push_back('_');
          compose_func_name.push_back(tf_op_name[i] + 'a' - 'A');
        } else {
          compose_func_name.push_back(tf_op_name[i]);
        }
      }
      return compose_func_name;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 6.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc

          lhs_shape.push_back(b);
          out_shape->push_back(b);
    
          lhs_reshape_segids.at(std::get<0>(i)) = num_lhs_reshape_segids++;
        }
      } else {
        const int64_t lhs_out_size =
            ProdShapeWithIndexInTuple<0>(lhs_type.getShape(), dnums.lhs_out);
        lhs_shape.push_back(lhs_out_size);
        out_shape->push_back(lhs_out_size);
    
        for (auto i : dnums.lhs_out) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/quantization/quantization_context.cc

          if (state.IsEmpty() && requantize.pos == RequantizeState::NO_REQUANTIZE) {
            input_specs.push_back(original_input_specs[i]);
          } else if (requantize.pos == RequantizeState::ON_OUTPUT) {
            input_specs.push_back(TypeAttr::get(requantize.params));
          } else {
            input_specs.push_back(TypeAttr::get(state.params));
          }
        }
        op->setAttr("input_specs", ArrayAttr::get(context, input_specs));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 08 01:38:03 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfrt/saved_model/saved_model.cc

                    func.getArgAttrOfType<mlir::StringAttr>(i, "tf.device")) {
              input_devices.push_back(input_device.getValue());
            } else {
              input_devices.push_back("");
            }
          }
          if (auto* bound_input =
                  mlir::tf_saved_model::LookupBoundInput(func, i, symbol_table)) {
            bound_inputs.push_back(bound_input);
          }
        }
    
        llvm::SmallVector<llvm::StringRef, 4> output_names;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  5. tensorflow/cc/framework/while_gradients_test.cc

            return s.status();
          },
          [](const Scope& s, const std::vector<Output>& inputs,
             std::vector<Output>* outputs) {
            outputs->push_back(ops::AddN(s, {inputs[0], inputs[1]}));
            outputs->push_back(ops::AddN(s, {inputs[1], 1}));
            outputs->push_back(inputs[2]);
            return s.status();
          });
      CreateBackprop();
    
      // The following execution traces illustrate why we expect dF/dj to be 5:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 27 20:32:17 UTC 2017
    - 7.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/utils/fake_session.cc

          t.flat<float>().setZero();
          outputs->push_back(t);
        } else if (output_name == "dense/kernel") {
          Tensor t =
              Tensor(tensorflow::DT_FLOAT, tensorflow::TensorShape({100, 50}));
          t.flat<float>().setZero();
          outputs->push_back(t);
        } else if (output_name == "var1") {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 03:47:51 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

                  static_cast<int64_t>(0)));
        }
        compilation_key_ops.push_back(compilation_key_op);
        core_to_compilation_key.push_back(compilation_key);
        core_to_device_ordinal_op.push_back(device_ordinal_op);
        if (device_cluster->getParentOfType<mlir::tf_device::ReplicateOp>())
          core_to_device_ordinal.push_back(
              core_to_device_ordinal_op[core]->getResults()[0]);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc

        for (auto j : dependency_graph.at(i)) {
          if (is_consumed_id[j]) continue;
          queue.push_back(j);
          is_consumed_id[j] = true;
        }
      }
    
      return success();
    }
    
    void FindProducers(Value start_node, std::vector<uint64_t> &neighbors) {
      llvm::DenseSet<Value> visited;
      std::vector<Value> queue;
      queue.push_back(start_node);
      visited.insert(start_node);
      while (!queue.empty()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc

                                                 flag.std_value(), inference_type));
            node_mins->push_back(min_max.first);
            node_maxs->push_back(min_max.second);
          } else {
            node_mins->push_back(std::nullopt);
            node_maxs->push_back(std::nullopt);
          }
        }
      }
    
      if (mlir::quant::GetInputNodeQuantSpecs(*node_names, *node_mins, *node_maxs,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:39:37 UTC 2024
    - 17.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc

          if (d == split_dim) {
            // Split dimension.
            slice_begin.push_back(begin);
            int64_t size = current_output_type.getDimSize(d);
            slice_size.push_back(size);
            begin += size;
          } else {
            slice_begin.push_back(0);
            // -1 means every elements.
            slice_size.push_back(-1);
          }
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 25.4K bytes
    - Viewed (0)
Back to top