Home
last modified time | relevance | path

Searched refs:FrameAndIter (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dcontrol_flow.h32 struct FrameAndIter { struct
36 FrameAndIter() {} in FrameAndIter() argument
38 FrameAndIter(uint64 frame, int64 iter) { in FrameAndIter() function
43 bool operator==(const FrameAndIter& other) const {
49 size_t operator()(const FrameAndIter& key) const { in operator()
51 CHECK_EQ(sizeof(uint64) + sizeof(int64), sizeof(FrameAndIter)); in operator()
52 return Hash64(reinterpret_cast<const char*>(&key), sizeof(FrameAndIter)); in operator()
Drendezvous.h58 const FrameAndIter& frame_iter);
Drendezvous_test.cc42 "/job:mnist/replica:1/task:2/device:GPU:0", "var0", FrameAndIter(0, 0)); in TEST()
101 name, FrameAndIter(0, 0)); in MakeKey()
Drendezvous.cc54 const FrameAndIter& frame_iter) { in CreateKey()
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc36 const FrameAndIter& frame_iter, string* key) { in GetRendezvousKey()
42 static FrameAndIter GetFrameAndIter(OpKernelContext* ctx, in GetFrameAndIter()
49 return FrameAndIter(reinterpret_cast<uint64>(ctx->call_frame()), 0); in GetFrameAndIter()
90 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in Compute()
91 if (frame_iter == FrameAndIter(0, 0)) { in Compute()
180 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in ComputeAsync()
181 if (frame_iter == FrameAndIter(0, 0)) { in ComputeAsync()
Dconstant_op_test.cc67 params.frame_iter = FrameAndIter(0, 0); in PersistentMemoryTrackingTest()
Drestore_op_test.cc173 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
418 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
Dsegment_reduction_ops_test.cc73 params.frame_iter = FrameAndIter(0, 0); in BM_SegmentReduction()
Dparse_tensor_test.cc57 params.frame_iter = FrameAndIter(0, 0); in ParseSerializedWithNodeDef()
Dops_testutil.h169 params_.get()->frame_iter = FrameAndIter(0, 0); in RunOpKernel()
Dsparse_to_dense_op_test.cc252 params.frame_iter = FrameAndIter(0, 0); in BM_SparseToDense()
Dnn_ops_test.cc767 params.frame_iter = FrameAndIter(0, 0); in BM_LRNFloat()
847 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPool()
952 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPoolBk()
1042 params.frame_iter = FrameAndIter(0, 0); in BM_MaxPool()
1218 params.frame_iter = FrameAndIter(0, 0); in BM_ReluFloat()
Drestore_v2_op_test.cc182 params.frame_iter = FrameAndIter(0, 0); in RunTest()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc90 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
110 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
144 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
175 "/job:mnist/replica:1/task:2/cpu:1", "foo", FrameAndIter(0, 0))); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_runner.cc136 tensor_name, FrameAndIter(0, 0)); in Run()
194 output_names[i], FrameAndIter(0, 0)); in Run()
Drendezvous_util_test.cc50 "/job:localhost/replica:0/task:0/device:GPU:0", name, FrameAndIter(0, 0)); in MakeStringKey()
Dkernel_benchmark_testlib.cc128 recv_device, tensor_name, FrameAndIter(0, 0)); in GetRendezvousKey()
Dprocess_function_library_runtime.cc120 target_device, name, FrameAndIter(0, 0)); in SendTensors()
139 target_device, name, FrameAndIter(0, 0)); in ReceiveTensorsAsync()
Ddirect_session.cc137 const FrameAndIter& frame_iter) { in GetRendezvousKey()
1332 input, device_set_.client_device()->attributes(), FrameAndIter(0, 0)); in CreateExecutors()
1338 FrameAndIter(0, 0)); in CreateExecutors()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc61 target, 1 /* src_incarnation */, target, in.name(), FrameAndIter(0, 0)); in ConstructFunctionGraph()
106 out.name(), FrameAndIter(0, 0)); in ConstructFunctionGraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc98 params.frame_iter = FrameAndIter(0, 0); in EvaluateNode()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc230 params.frame_iter = FrameAndIter(0, 0); in RunAsync()
Ddataset_test_base.cc184 params_->frame_iter = FrameAndIter(0, 0); in CreateOpKernelContext()
Dsingle_threaded_executor_test.cc135 name, FrameAndIter(0, 0)), in Key()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc233 params.frame_iter = FrameAndIter(0, 0); in Run()

12