GatherNd - Code Search

tensorflow/cc/gradients/array_grad_test.cc

  TensorShape y_shape({2});
  auto y = GatherNd(scope_, x, indices);
  RunTest(x, x_shape, y, y_shape);
}

TEST_F(ArrayGradTest, GatherNdGrad_SliceIndexing) {
  TensorShape shape({2, 2});
  auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape));
  auto indices = Const(scope_, {{1}, {0}});
  auto y = GatherNd(scope_, x, indices);
  RunTest(x, shape, y, shape);
}

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Tue Oct 10 23:33:32 UTC 2023

- 19.3K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/cc/gradients/array_grad.cc

  grad_outputs->push_back(ScatterNd(scope, indices, grad_inputs[0], ref_shape));
  grad_outputs->push_back(NoGradient());
  return scope.status();
}
REGISTER_GRADIENT_OP("GatherNd", GatherNdGrad);

Status CheckNumericsGrad(const Scope& scope, const Operation& op,
                         const std::vector<Output>& grad_inputs,
                         std::vector<Output>* grad_outputs) {

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Tue Oct 10 23:33:32 UTC 2023

- 31.7K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc

    return false;
  }

  // TODO(unda, 253431636): constrained kernel
  std::set<string> constrained_kernel = {"Diag",
                                         "DiagPart",
                                         "GatherNd",
                                         "GatherV2",
                                         "QuantizeAndDequantizeV2",
                                         "QuantizeAndDequantizeV3",

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Sat Jan 27 16:26:51 UTC 2024

- 13K bytes

- Viewed (0)

github.com/gradle/gradle

platforms/core-configuration/model-core/src/main/java/org/gradle/internal/reflect/annotations/impl/DefaultTypeAnnotationMetadataStore.java

    /**
     * Constructs the store.
     *
     * @param recordedTypeAnnotations Annotations on the type itself that should be gathered.

Registered: Wed Jun 12 18:38:38 UTC 2024

- Last Modified: Mon May 06 21:54:36 UTC 2024

- 37.6K bytes

- Viewed (0)

github.com/istio/istio

tests/integration/telemetry/policy/helper_test.go

	PortName string
	HTTP2    bool
	Host     string
	Expected Expected
}

// Expected contains the metric and query to run against
// prometheus to validate that expected telemetry information was gathered;
// as well as the http response code
type Expected struct {
	Query           prometheus.Query
	StatusCode      int
	Metric          string
	PromQueryFormat string
	Protocol        string

Registered: Fri Jun 14 15:00:06 UTC 2024

- Last Modified: Thu May 02 21:29:40 UTC 2024

- 11.4K bytes

- Viewed (0)

github.com/golang/go

src/internal/trace/traceviewer/http.go

var templMain = template.Must(template.New("").Parse(`
<html>
<style>` + CommonStyle + `</style>
<body>
<h1>cmd/trace: the Go trace event viewer</h1>
<p>
  This web server provides various visualizations of an event log gathered during
  the execution of a Go program that uses the <a href='https://pkg.go.dev/runtime/trace'>runtime/trace</a> package.
</p>

<h2>Event timelines for running goroutines</h2>
{{range $i, $view := $}}

Registered: Wed Jun 12 16:32:35 UTC 2024

- Last Modified: Tue Nov 21 21:29:53 UTC 2023

- 12.6K bytes

- Viewed (0)

github.com/tensorflow/tensorflow

tensorflow/compiler/aot/tfcompile.bzl

        full set of deps needed by the generated library.
      enable_xla_hlo_profiling: Enable XLA HLO profiling in the generated
        program, and emit metadata that lets us pretty-print the gathered
        profile counters.
      enable_tracemes: Tell tfcompile to generate calls to
        TraceMe::Activity{Start|End} around HLO instructions that can be used by
        Xprof to construct profiler timelines.

Registered: Sun Jun 16 05:45:23 UTC 2024

- Last Modified: Thu May 02 19:18:08 UTC 2024

- 21.8K bytes

- Viewed (0)

github.com/golang/go

src/cmd/vendor/golang.org/x/sys/unix/syscall_unix.go

}

// SendmsgBuffers sends a message on a socket to an address using the sendmsg
// system call. This function is equivalent to SendmsgN, but the non-control
// data is gathered from buffers.
func SendmsgBuffers(fd int, buffers [][]byte, oob []byte, to Sockaddr, flags int) (n int, err error) {
	iov := make([]Iovec, len(buffers))
	for i := range buffers {
		if len(buffers[i]) > 0 {

Registered: Wed Jun 12 16:32:35 UTC 2024

- Last Modified: Thu Oct 19 23:33:33 UTC 2023

- 16.5K bytes

- Viewed (0)

github.com/golang/go

src/cmd/compile/internal/types2/object.go

	// 2) internally created but not yet type-checked.
	// For case 1) we can't do anything; the client must know what they are doing.
	// For case 2) we can use the information gathered by the resolver.
	return obj.hasPtrRecv_
}

func (*Func) isDependency() {} // a function may be a dependency of an initialization expression

// A Label represents a declared label.

Registered: Wed Jun 12 16:32:35 UTC 2024

- Last Modified: Tue May 07 20:08:23 UTC 2024

- 20.1K bytes

- Viewed (0)

github.com/golang/go

src/go/types/object.go

	// 2) internally created but not yet type-checked.
	// For case 1) we can't do anything; the client must know what they are doing.
	// For case 2) we can use the information gathered by the resolver.
	return obj.hasPtrRecv_
}

func (*Func) isDependency() {} // a function may be a dependency of an initialization expression

// A Label represents a declared label.

Registered: Wed Jun 12 16:32:35 UTC 2024

- Last Modified: Tue May 07 20:08:23 UTC 2024

- 20.1K bytes

- Viewed (0)

Search Options