| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 41 const FrameAndIter& frame_iter, string* key) { in GetRendezvousKey() argument
43 strings::StrAppend(key, key_prefix, ";", frame_iter.frame_id, ":", in GetRendezvousKey()
44 frame_iter.iter_id); in GetRendezvousKey()
56 return ctx->frame_iter(); in GetFrameAndIter()
95 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in Compute() local
96 if (frame_iter == FrameAndIter(0, 0)) { in Compute()
105 GetRendezvousKey(key_prefix_, frame_iter, &in_loop_parsed.buf_); in Compute()
207 FrameAndIter frame_iter = GetFrameAndIter(ctx, hostmem_sendrecv_); in ComputeAsync() local
208 if (frame_iter == FrameAndIter(0, 0)) { in ComputeAsync()
215 GetRendezvousKey(key_prefix_, frame_iter, &in_loop_parsed.buf_); in ComputeAsync()
|
| D | variable_ops.cc | 110 auto unique_name = TemporaryVariableName(var_name_, context->frame_iter()); in Compute()
163 auto unique_name = TemporaryVariableName(var_name_, context->frame_iter()); in Compute()
|
| D | nccl_ops.cc | 52 c->frame_iter().frame_id, ":", in GetCollectiveKey()
53 c->frame_iter().iter_id); in GetCollectiveKey()
|
| D | segment_reduction_ops_test.cc | 77 params.frame_iter = FrameAndIter(0, 0); in BM_UnsortedSegmentReduction()
140 params.frame_iter = FrameAndIter(0, 0); in BM_SegmentReduction()
|
| D | constant_op_test.cc | 71 params.frame_iter = FrameAndIter(0, 0); in PersistentMemoryTrackingTest()
|
| D | restore_op_test.cc | 176 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
421 params.frame_iter = FrameAndIter(0, 0); in TEST_F()
|
| D | parse_tensor_test.cc | 60 params.frame_iter = FrameAndIter(0, 0); in ParseSerializedWithNodeDef()
|
| D | nn_ops_test.cc | 808 params.frame_iter = FrameAndIter(0, 0); in BM_LRNFloat()
887 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPool()
990 params.frame_iter = FrameAndIter(0, 0); in BM_AvgPoolBk()
1081 params.frame_iter = FrameAndIter(0, 0); in BM_MaxPool()
1263 params.frame_iter = FrameAndIter(0, 0); in BM_ReluFloat()
1334 params.frame_iter = FrameAndIter(0, 0); in BM_SoftplusFloat()
|
| D | sparse_to_dense_op_test.cc | 250 params.frame_iter = FrameAndIter(0, 0); in BM_SparseToDense()
|
| D | ops_testutil.cc | 168 params_->frame_iter = FrameAndIter(0, 0); in RunOpKernel()
|
| D | restore_v2_op_test.cc | 185 params.frame_iter = FrameAndIter(0, 0); in RunTest()
|
| D | collective_ops.cc | 44 ctx->frame_iter().frame_id, ":", in CollectiveKey()
45 ctx->frame_iter().iter_id); in CollectiveKey()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 56 const FrameAndIter& frame_iter) { in CreateKey() argument
67 dst_device, ";", name, ";", frame_iter.frame_id, ":", frame_iter.iter_id); in CreateKey()
|
| D | rendezvous.h | 176 const FrameAndIter& frame_iter);
|
| D | op_kernel.h | 653 FrameAndIter frame_iter; member
1143 FrameAndIter frame_iter() const { return params_->frame_iter; } in frame_iter() function
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 433 VLOG(1) << "TpuInfeedEnqueueOp::DoWork. iter_id=" << ctx->frame_iter().iter_id in DoWork()
463 << ctx->frame_iter().iter_id << " device_ordinal=" << device_ordinal; in DoWork()
494 << ctx->frame_iter().iter_id << " device_ordinal=" << device_ordinal; in DoWork()
538 << ctx->frame_iter().iter_id << " device_ordinal=" << device_ordinal; in DoWork()
|
| D | tpu_compile_op_common.cc | 388 ctx->frame_iter().frame_id, ":", ctx->frame_iter().iter_id, ":"); in ComputeInternal()
|
| /external/tensorflow/tensorflow/core/activity_watcher/ |
| D | activity_utils.cc | 34 {"iter_num", absl::StrCat(context->frame_iter().iter_id)}, in ActivityFromContext()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 898 VLOG(3) << "IteratorGetNextOp enter. iter_id=" << ctx->frame_iter().iter_id; in DoCompute()
900 VLOG(3) << "IteratorGetNextOp exit. iter_id=" << ctx->frame_iter().iter_id; in DoCompute()
911 {{"id", ctx->step_id()}, {"iter_num", ctx->frame_iter().iter_id}}); in DoCompute()
937 << ctx->frame_iter().iter_id; in DoCompute()
940 << ctx->frame_iter().iter_id; in DoCompute()
951 {{"id", ctx->step_id()}, {"iter_num", ctx->frame_iter().iter_id}}); in DoCompute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 77 params.frame_iter = FrameAndIter(0, 0); in EvaluateNode()
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | eval_utils.cc | 124 params.frame_iter = tensorflow::FrameAndIter(0, 0); in EvaluateOperation()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 639 ->frame_iter() in TF_GetFrameId()
651 ->frame_iter() in TF_GetIterId()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 272 params.frame_iter = FrameAndIter(0, 0); in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 333 params.frame_iter = FrameAndIter(0, 0); in Run()
|
| D | direct_session.cc | 152 const FrameAndIter& frame_iter) { in GetRendezvousKey() argument
156 frame_iter.frame_id, ":", frame_iter.iter_id); in GetRendezvousKey()
|