/external/tensorflow/tensorflow/core/kernels/ |
D | sendrecv_ops.cc | 36 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()
|
D | nccl_ops.cc | 48 c->frame_iter().frame_id, ":", in GetCollectiveKey() 49 c->frame_iter().iter_id); in GetCollectiveKey()
|
D | constant_op_test.cc | 67 params.frame_iter = FrameAndIter(0, 0); in PersistentMemoryTrackingTest()
|
D | collective_ops.cc | 32 c->frame_iter().frame_id, ":", in GetCollectiveKey() 33 c->frame_iter().iter_id); in GetCollectiveKey()
|
D | restore_op_test.cc | 173 params.frame_iter = FrameAndIter(0, 0); in TEST_F() 418 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
|
D | segment_reduction_ops_test.cc | 73 params.frame_iter = FrameAndIter(0, 0); in BM_SegmentReduction()
|
D | parse_tensor_test.cc | 57 params.frame_iter = FrameAndIter(0, 0); in ParseSerializedWithNodeDef()
|
D | ops_testutil.h | 169 params_.get()->frame_iter = FrameAndIter(0, 0); in RunOpKernel()
|
D | sparse_to_dense_op_test.cc | 252 params.frame_iter = FrameAndIter(0, 0); in BM_SparseToDense()
|
D | nn_ops_test.cc | 767 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()
|
D | restore_v2_op_test.cc | 182 params.frame_iter = FrameAndIter(0, 0); in RunTest()
|
/external/tensorflow/tensorflow/core/framework/ |
D | rendezvous.cc | 54 const FrameAndIter& frame_iter) { in CreateKey() argument 65 dst_device, ";", name, ";", frame_iter.frame_id, ":", frame_iter.iter_id); in CreateKey()
|
D | rendezvous.h | 58 const FrameAndIter& frame_iter);
|
D | op_kernel.h | 663 FrameAndIter frame_iter; member 1150 FrameAndIter frame_iter() const { return params_->frame_iter; } in frame_iter() function
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | evaluation_utils.cc | 98 params.frame_iter = FrameAndIter(0, 0); in EvaluateNode()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | single_threaded_executor.cc | 230 params.frame_iter = FrameAndIter(0, 0); in RunAsync()
|
D | dataset_test_base.cc | 184 params_->frame_iter = FrameAndIter(0, 0); in CreateOpKernelContext()
|
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | kernel_and_device.cc | 233 params.frame_iter = FrameAndIter(0, 0); in Run()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | direct_session.cc | 137 const FrameAndIter& frame_iter) { in GetRendezvousKey() argument 141 frame_iter.frame_id, ":", frame_iter.iter_id); in GetRendezvousKey()
|
D | executor.cc | 1732 params.frame_iter = FrameAndIter(input_frame->frame_id, input_iter); in Process()
|