Home
last modified time | relevance | path

Searched refs:frame_iter (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc36 const FrameAndIter& frame_iter, string* key) { in GetRendezvousKey() argument
38 strings::StrAppend(key, key_prefix, ";", frame_iter.frame_id, ":", in GetRendezvousKey()
39 frame_iter.iter_id); in GetRendezvousKey()
51 return ctx->frame_iter(); in GetFrameAndIter()
90 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in Compute() local
91 if (frame_iter == FrameAndIter(0, 0)) { in Compute()
99 GetRendezvousKey(key_prefix_, frame_iter, &in_loop_parsed.buf_); in Compute()
180 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in ComputeAsync() local
181 if (frame_iter == FrameAndIter(0, 0)) { in ComputeAsync()
187 GetRendezvousKey(key_prefix_, frame_iter, &in_loop_parsed.buf_); in ComputeAsync()
Dnccl_ops.cc48 c->frame_iter().frame_id, ":", in GetCollectiveKey()
49 c->frame_iter().iter_id); in GetCollectiveKey()
Dconstant_op_test.cc67 params.frame_iter = FrameAndIter(0, 0); in PersistentMemoryTrackingTest()
Dcollective_ops.cc32 c->frame_iter().frame_id, ":", in GetCollectiveKey()
33 c->frame_iter().iter_id); in GetCollectiveKey()
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/framework/
Drendezvous.cc54 const FrameAndIter& frame_iter) { in CreateKey() argument
65 dst_device, ";", name, ";", frame_iter.frame_id, ":", frame_iter.iter_id); in CreateKey()
Drendezvous.h58 const FrameAndIter& frame_iter);
Dop_kernel.h663 FrameAndIter frame_iter; member
1150 FrameAndIter frame_iter() const { return params_->frame_iter; } in frame_iter() function
/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()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc233 params.frame_iter = FrameAndIter(0, 0); in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc137 const FrameAndIter& frame_iter) { in GetRendezvousKey() argument
141 frame_iter.frame_id, ":", frame_iter.iter_id); in GetRendezvousKey()
Dexecutor.cc1732 params.frame_iter = FrameAndIter(input_frame->frame_id, input_iter); in Process()