| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/ |
| D | gradient-stats.h | 50 auto me_flat = t.unaligned_flat<float>();
51 auto other_flat = other.t.unaligned_flat<float>();
65 auto me_flat = t.unaligned_flat<float>();
66 auto other_flat = other.t.unaligned_flat<float>();
74 auto me_flat = t.unaligned_flat<float>();
82 auto me_flat = t.unaligned_flat<float>(); in IsZero()
93 auto me_flat = t.unaligned_flat<float>();
105 auto me_flat = t.unaligned_flat<float>(); in Magnitude()
|
| D | node-stats.h | 73 g = grad_stats.first.t.unaligned_flat<float>()(0); in NodeStats()
74 h = grad_stats.second.t.unaligned_flat<float>()(0); in NodeStats()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils_test.cc | 35 EXPECT_EQ(WeightedGiniImpurity(vals.unaligned_flat<float>()), 6); in TEST()
38 EXPECT_EQ(WeightedGiniImpurity(zero.unaligned_flat<float>()), 0); in TEST()
46 EXPECT_FLOAT_EQ(WeightedVariance(sums.unaligned_flat<float>(), in TEST()
47 squares.unaligned_flat<float>(), 3), in TEST()
51 EXPECT_FLOAT_EQ(WeightedVariance(zero.unaligned_flat<float>(), in TEST()
52 zero.unaligned_flat<float>(), 1), in TEST()
|
| D | tree_utils.cc | 94 total_counts.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestClassification()
99 split_counts.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestClassification()
167 total_sums.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
169 total_squares.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
171 split_sums.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
173 split_squares.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
331 .unaligned_flat<float>()(0); in BestSplitDominatesClassificationHoeffding()
514 .unaligned_flat<float>()(0); in BestSplitDominatesClassificationChebyshev()
|
| D | reinterpret_string_to_float_op.cc | 40 auto out_data = output_data.unaligned_flat<float>(); in Evaluate()
41 const auto in_data = input_data.unaligned_flat<string>(); in Evaluate()
|
| D | stats_ops.cc | 155 const auto leaf_ids = leaf_ids_tensor.unaligned_flat<int32>(); in UpdateStats()
270 const auto leaf_ids = leaf_ids_tensor.unaligned_flat<int32>(); in Compute()
345 auto output = output_finished_t->unaligned_flat<int32>(); in Compute()
379 const auto finished = finished_nodes.unaligned_flat<int32>(); in Compute()
|
| D | tree_utils.h | 54 counts.unaligned_flat<T>().sum(); in Sum()
151 auto flat = counts.unaligned_flat<T>();
|
| D | model_ops.cc | 228 auto out_paths = output_tree_paths->unaligned_flat<string>(); in Compute()
362 const auto leaves = leaf_ids.unaligned_flat<int32>(); in UpdateModel()
401 auto counts = output_counts->unaligned_flat<int32>(); in Compute()
|
| D | scatter_add_ndim_op.cc | 67 const auto deltas = deltas_tensor.unaligned_flat<float>(); in Compute()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| D | utils.cc | 30 const auto p = point.unaligned_flat<float>(); in LeftProbability()
31 const auto w = weight.unaligned_flat<float>(); in LeftProbability()
45 const auto p = point.unaligned_flat<float>(); in LeftProbabilityK()
46 const auto w = weight.unaligned_flat<float>(); in LeftProbabilityK()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | candidate_graph_runner.cc | 100 return outputs[0].unaligned_flat<float>()(0); in SplitScore()
106 ParseProtoUnlimited(node, outputs[0].unaligned_flat<string>()(0)); in GetSplit()
118 const auto& counts = outputs[0].unaligned_flat<float>(); in GetLeftStats()
128 const auto& counts = outputs[0].unaligned_flat<float>(); in GetRightStats()
|
| D | input_target.h | 58 new SingleDimStorageType(target.unaligned_flat<float>()), in TensorInputTarget()
59 new SingleDimStorageType(weight.unaligned_flat<float>()), in TensorInputTarget()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_util.cc | 46 tmp.unaligned_flat<string>() = other.unaligned_flat<string>(); in DeepCopy()
49 tmp.unaligned_flat<Variant>() = other.unaligned_flat<Variant>(); in DeepCopy()
|
| D | tensor_util_test.cc | 147 EXPECT_EQ(2, y.unaligned_flat<int32>()(i)); in TEST()
177 EXPECT_EQ("goodbye", y.unaligned_flat<string>()(i)); in TEST()
209 y.unaligned_flat<Variant>()(i).get<Tensor>()->scalar<string>()()); in TEST()
|
| D | tensor_test.cc | 1263 y.unaligned_flat<float>().setConstant(1.0); in TEST()
1265 EXPECT_EQ(1.0, y.unaligned_flat<float>()(i)); in TEST()
1289 auto tz = z.unaligned_flat<float>(); in TEST()
1302 y.unaligned_flat<float>().setConstant(1.0); in TEST()
1304 EXPECT_EQ(1.0, y.unaligned_flat<float>()(i)); in TEST()
|
| D | tensor.h | 402 typename TTypes<T>::UnalignedFlat unaligned_flat() { in unaligned_flat() function
491 typename TTypes<T>::UnalignedConstFlat unaligned_flat() const { in unaligned_flat() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | non_max_suppression_op.cc | 265 float* scores_data = per_batch_scores.unaligned_flat<float>().data(); in BatchedNonMaxSuppressionOp()
266 float* boxes_data = per_batch_boxes.unaligned_flat<float>().data(); in BatchedNonMaxSuppressionOp()
294 boxes.unaligned_flat<float>().data()); in BatchedNonMaxSuppressionOp()
|
| D | list_kernels.h | 559 tmp.unaligned_flat<T>(); in Compute()
708 tmp.unaligned_flat<T>(); in Compute()
734 tmp.unaligned_flat<T>(); in Scatter()
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.cc | 764 original.unaligned_flat<T>()); in FinishTimeStep()
802 functor::TensorCopyUnaligned<Device, T>()(device_, t.unaligned_flat<T>(), in AlignTensor()
995 cs_tensor.unaligned_flat<T>()); in Compute()
997 h_tensor.unaligned_flat<T>()); in Compute()
1274 device, x_grad_tensor.unaligned_flat<T>()); in Compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer_test.cc | 318 auto alias = actual.template unaligned_flat<T>(); in RunTest()
|
| D | ring_reducer_test.cc | 338 auto alias = actual.template unaligned_flat<T>(); in RunTest()
|